Skip to content

Latest commit

 

History

History
20 lines (14 loc) · 720 Bytes

readme.md

File metadata and controls

20 lines (14 loc) · 720 Bytes

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