We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents d30dcb7 + dc35038 commit 80e8177Copy full SHA for 80e8177
README.md
@@ -93,8 +93,7 @@ For example:
93
python -m tensorflow.python.tools.freeze_graph \
94
--input_graph=my_checkpoint_dir/graphdef.pb \
95
--input_binary=true \
96
- --input_names=input:0 \
97
- --output_node_names=output:0 \
+ --output_node_names=output \
98
--input_checkpoint=my_checkpoint_dir \
99
--output_graph=tests/models/fc-layers/frozen.pb
100
```
tests/run_pretrained_models.yaml
@@ -9,14 +9,6 @@ benchtf-fc:
9
outputs:
10
- output:0
11
12
-benchtf-fc:
13
- model: tests/models/fc-layers/frozen.pb
14
- input_get: get_ramp
15
- inputs:
16
- "X:0": [1, 784]
17
- outputs:
18
- - output:0
19
-
20
benchtf-conv:
21
model: tests/models/conv-layers/frozen.pb
22
input_get: get_ramp
0 commit comments