Skip to content

Commit 80e8177

Browse files
authored
Merge pull request #69 from pengwa/few-fixes
fix minor issues in test and freeze command
2 parents d30dcb7 + dc35038 commit 80e8177

File tree

2 files changed

+1
-10
lines changed

2 files changed

+1
-10
lines changed

README.md

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -93,8 +93,7 @@ For example:
9393
python -m tensorflow.python.tools.freeze_graph \
9494
--input_graph=my_checkpoint_dir/graphdef.pb \
9595
--input_binary=true \
96-
--input_names=input:0 \
97-
--output_node_names=output:0 \
96+
--output_node_names=output \
9897
--input_checkpoint=my_checkpoint_dir \
9998
--output_graph=tests/models/fc-layers/frozen.pb
10099
```

tests/run_pretrained_models.yaml

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -9,14 +9,6 @@ benchtf-fc:
99
outputs:
1010
- output:0
1111

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-
2012
benchtf-conv:
2113
model: tests/models/conv-layers/frozen.pb
2214
input_get: get_ramp

0 commit comments

Comments
 (0)