浏览代码

the bytes file will besaved under the name of the environment, not its path

/tag-0.2.0
vincentpierre 7 年前
当前提交
d01bd6c2
共有 1 个文件被更改,包括 4 次插入1 次删除
  1. 5
      python/ppo.py

5
python/ppo.py


if steps != 0 and train_model:
save_model(sess, model_path=model_path, steps=steps, saver=saver)
env.close()
export_graph(model_path, env_name)
graph_name = (env_name.strip()
.replace('.app', '').replace('.exe', '').replace('.x86_64', '').replace('.x86', ''))
graph_name = os.path.basename(os.path.normpath(graph_name))
export_graph(model_path, graph_name)
正在加载...
取消
保存