【发布时间】:2018-04-06 08:36:42
【问题描述】:
我在装有 High Sierra 的 Mac 上使用 TensorFlow 1.7 和 Python 3.6.5。
我已经训练了我的第一个 MNIST 模型,所以我基本上有
- 具有 CNN 图结构的 graph.pbtxt 文件
- 一些 model.ckpt-21000 文件(.meta、.index .data)
我尝试在我的 bash 上使用命令行 freeze_graph 命令冻结图形:
freeze_graph
--input_graph=/…/graph.pbtxt
--input_checkpoint=/…/model.ckpt-21000
--input_binary=false
--output_graph=/…/frozen_mnist.pb
--output_node_names=softmax_tensor
但是我收到了这个错误:
Traceback (most recent call last):
File “/usr/local/bin/freeze_graph”, line 11, in <module>
sys.exit(main())
TypeError: main() missing 1 required positional argument: ‘unused_args’
我不确定我在那里缺少什么。 我很确定我使用了正确的语法。
【问题讨论】:
标签: python tensorflow mnist