From 78d6627bc619235bc8f1605a919f74b3540b05f8 Mon Sep 17 00:00:00 2001 From: Jason Date: Thu, 14 Mar 2019 13:28:31 +0800 Subject: [PATCH] Update README.md --- tensorflow2fluid/README.md | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tensorflow2fluid/README.md b/tensorflow2fluid/README.md index a0cfe42..8159259 100644 --- a/tensorflow2fluid/README.md +++ b/tensorflow2fluid/README.md @@ -40,6 +40,7 @@ python src/convert.py --pb_file tf_model.pb \ |input_shape|输入tensor的shape(batch维度以None表示)，shape之间以空格分隔，shape内各维度以逗号分隔| |input_format|输入数据格式，NHWC/NCHW/OTHER| |output_nodes|输出tensor名，多个输出时以空格分隔| +|use_cuda|转换过程中是否使用GPU，默认True| |save_dir|转换后的模型保存路径| 目前支持tensorflow保存的checkpoint模型和将参数及模型结构序列化存储的pb模型，前者须指定meta_file和ckpt_dir，后者则指定pb_file @@ -48,12 +49,12 @@ python src/convert.py --pb_file tf_model.pb \ 文件|作用 :------------------:|:-----------------------------------------------: my_model.py|基于PaddlePaddle实现的模型网络结构python代码 -ref_name.txt|my_model.py中各tensor与原TensorFlow模型中的tensor对应关系 +ref_name.info|my_model.py中各tensor与原TensorFlow模型中的tensor对应关系 const_\*/params_\*|转换后的模型参数文件 save_var.list|模型载入过程中的变量list ### 加载转换后的模型并用于预测 -本目录下提示的[paddle_model_loader.py](paddle_model_loader.py)可用于加载转换后的模型 +本目录下的[model_loader.py](tf2fluid/model_loader.py)可用于加载转换后的模型 使用转换后的模型主要注意，**模型转换后，计算结果与原模型存在一定精度的diff，因此务必检查模型转换前后，在输入同样的数据前提下,diff是否符合预期** -- GitLab