GitHub - KangJinle/java-bert-predict: turn bert pretrain checkpoint into saved model for a feature extracting demo in java · GitHub
Skip to content

About

turn bert pretrain checkpoint into saved model for a feature extracting demo in java

Resources

Stars

0 stars

Watchers

1 watching

Forks

 
 

Latest commit

 

History

6 Commits

Folders and files

Repository files navigation

java-bert-predict

turn bert pretrain checkpoint into saved model for a feature extracting demo in java

usage

  1. download google bert pretrain from here and unzip it into a proper path.

  2. because java cannot work with tensorflow checkpoint directly, we need to transform checkpoint into saved_model with following command

    python script/checkpoint_to_saved_model.py path/to/unzipped/pretrain/checkpoint path/to/save/model

  3. run java program with path/to/save/model as params

About

turn bert pretrain checkpoint into saved model for a feature extracting demo in java

Resources

Stars

0 stars

Watchers

1 watching

Forks

Releases

Packages

Contributors

Languages