diff --git a/tensorflow2fluid/README.md b/tensorflow2fluid/README.md index 803f9dde3fd87e2c97b5a4bbc39ce844e1d7391b..c559ccc1dec02c699272f530e95c9e73d32937cf 100644 --- a/tensorflow2fluid/README.md +++ b/tensorflow2fluid/README.md @@ -43,7 +43,7 @@ python src/convert.py --pb_file tf_model.pb \ |output_nodes|输出tensor名,多个输出时以空格分隔| |save_dir|转换后的模型保存路径| -目前支持tensorflow保存的checkpoint模型和将参数及模型结构序列化存储的pb模型,前者须指定meta_file和ckpt_file,后者则指定pb_file +目前支持tensorflow保存的checkpoint模型和将参数及模型结构序列化存储的pb模型,前者须指定meta_file和ckpt_dir,后者则指定pb_file ### 转换后模型文件说明 文件|作用