Использование инструментов преобразования Tensorflow Graph
Когда я пытаюсь использовать инструмент преобразования графиков с FrozenGraph в Windows 10 через командную строку, я всегда получаю одну и ту же ошибку. Код:
C://tensorflow/bazel-bin/tensorflow/tools/graph_transforms/transform_graph ^
--in_graph=C://tensorflow-for-poets-2/tf_files/retrained_graph.pb ^
--out_graph=C://tensorflow-for-poets-2/new_optimized.pb ^
--inputs=Mul ^
--outputs=final_result ^
--transforms=' ^
strip_unused_nodes ^
fold_constants(ignore_errors=true) ^
fold_batch_norms ^
fold_old_batch_norms'
Ошибка:
2018-07-27 14:59:09.303184: E tensorflow/tools/graph_transforms/transform_graph.cc:217] Unknown argument strip_unused_nodes.
usage: C://tensorflow/bazel- bin/tensorflow/tools/graph_transforms/transform_graph
Как мне поступить? Мне нужно оптимизировать этот график для работы с tenorflowjs. Это InceptionV3, поэтому он довольно большой для мобильных устройств.
1 ответ
Решение
Неизвестный аргумент strip_unused_nodes
Ошибка указывает на то, что аргументы отсутствуют strip_unused_nodes
, Вам необходимо указать тип и форму ваших входных данных. Вы можете взглянуть на документы