Question

我正在尝试冻结多GPU TF估计器输出的检查点的图形。这是我使用的命令。

python -m tensorflow.python.tools.freeze_graph --input_graph models/graph.pbtxt --input_checkpoint models/model.ckpt-40886 --output_graph hello.pb --output_node_names Decoder/Layer8/filter --input_binary=false

出现以下MemoryError错误。 graph.pbtxt文件约为1.5GB，我的机器具有64Gigs的RAM。

  .....
  .....
  .....
  File "/home/will/pyenv3/lib/python3.6/site-packages/google/protobuf/text_format.py", line 837, in _MergeMessageField
self._MergeField(tokenizer, sub_message)
  File "/home/will/pyenv3/lib/python3.6/site-packages/google/protobuf/text_format.py", line 749, in _MergeField
tokenizer.Consume(':')
  File "/home/will/pyenv3/lib/python3.6/site-packages/google/protobuf/text_format.py", line 1100, in Consume
if not self.TryConsume(token):
  File "/home/will/pyenv3/lib/python3.6/site-packages/google/protobuf/text_format.py", line 1087, in TryConsume
self.NextToken()
  File "/home/will/pyenv3/lib/python3.6/site-packages/google/protobuf/text_format.py", line 1336, in NextToken
match = self._TOKEN.match(self._current_line, self._column)
  MemoryError

是否有任何方法可以冻结模型以在TF Estimator的训练检查点输出上工作？

来自TF.Estimator的.pbtxt的Frozen_graph MemoryError

0 个答案: