Skip to content

Convert your ckpt to pb savedModel for converting to tensorflow.js

Notifications You must be signed in to change notification settings

r1cebank/tf-ckpt-2-pb

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 

Repository files navigation

tensorflow ckpt to pb

Wrote a simple cli application that will convert saved tensorflow checkpoint to pb savedModel.

Reason behind this is sometimes I am writing tensorflow.js programs and some of the models saved by checkpoint needs to be converted to SavedModel, nothing fancy just a simple tool i use for myself

Usage:

python convert.py --checkpoint XX --model XX --out-path ./export

--checkpoint

The file path of your checkpoint

--model

The exported MetaGraph

--out

The output directory, needs to be none existant

The outputed saved model will have the tag tfckpt2pb

About

Convert your ckpt to pb savedModel for converting to tensorflow.js

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages