freeze_graph MemoryError для .pbtxt из TF.Estimator - PullRequest
Новая
       6

freeze_graph MemoryError для .pbtxt из TF.Estimator

0 голосов
/ 18 января 2019

Я пытаюсь заморозить график для контрольных точек, которые были выведены с помощью Multi-GPU TF Estimator. Вот команда, которую я использовал. 

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,5 ГБ, а на моей машине установлено 64 ГБ ОЗУ. 

  .....
  .....
  .....
  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?

...