浏览代码

Merge pull request #876 from Unity-Technologies/release-windows-save-model-fix

some hack to make windows save the model when do ctrl+c
/develop-generalizationTraining-TrainerController
GitHub 6 年前
当前提交
9156737e
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1
      python/unitytrainers/trainer_controller.py

1
python/unitytrainers/trainer_controller.py


if global_step != 0 and self.train_model:
self._save_model(sess, steps=global_step, saver=saver)
except KeyboardInterrupt:
print('--------------------------Now saving model-------------------------')
if self.train_model:
self.logger.info("Learning was interrupted. Please wait while the graph is generated.")
self._save_model(sess, steps=global_step, saver=saver)

正在加载...
取消
保存