提交 f23282dc 编写于 作者: P Peng Li

fix bug in saving model (invoke the wrong function)

上级 81207201
...@@ -111,7 +111,7 @@ class DeepSpeech2Model(object): ...@@ -111,7 +111,7 @@ class DeepSpeech2Model(object):
output_model_path = os.path.join(output_model_dir, output_model_path = os.path.join(output_model_dir,
"params.latest.tar.gz") "params.latest.tar.gz")
with gzip.open(output_model_path, 'w') as f: with gzip.open(output_model_path, 'w') as f:
self._parameters.to_tar(f) trainer.save_parameter_to_tar(f)
print("\nPass: %d, Batch: %d, TrainCost: %f" % print("\nPass: %d, Batch: %d, TrainCost: %f" %
(event.pass_id, event.batch_id + 1, (event.pass_id, event.batch_id + 1,
cost_sum / cost_counter)) cost_sum / cost_counter))
...@@ -136,7 +136,7 @@ class DeepSpeech2Model(object): ...@@ -136,7 +136,7 @@ class DeepSpeech2Model(object):
output_model_path = os.path.join( output_model_path = os.path.join(
output_model_dir, "params.pass-%d.tar.gz" % event.pass_id) output_model_dir, "params.pass-%d.tar.gz" % event.pass_id)
with gzip.open(output_model_path, 'w') as f: with gzip.open(output_model_path, 'w') as f:
self._parameters.to_tar(f) trainer.save_parameter_to_tar(f)
# run train # run train
trainer.train( trainer.train(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册