Model 在OPENAI基线中保存模型

Openai基线使用以下命令保存经过训练的模型

python -m baselines.run --alg=ppo2 --env=PongNoFrameskip-v4 --num_timesteps=2e7 --save_path=~/models/pong_20M_ppo2

但是保存的训练模型不是以

.ckpt.meta
.ckpt.index
.ckpt.data
checkpoint

---

在早期版本中是以这种形式出现的。我们如何将模型保存为.ckpt.meta、.ckpt.index、.ckpt.data和检查点格式？

我遇到了同样的问题，我通过对基线代码进行一些调整来解决这个问题

基线中有两对用于保存和加载模型的方法（save_state&load_state对和save_variables&loas_variables对），您可以在baselines/common/tf_util.py（line325~line372）中看到

对于最新版本的基线，以.ckpt.meta、.ckpt.index、.ckpt.data和检查点格式保存和加载模型的save_state和load_state对已被放弃，因此您需要重新启用save_state和load_state对

以ppo2为例，在baselines/ppo2/model.py中，进行以下替换： 在第125行中，更换

self.save = functools.partial(save_variables, sess=sess)
self.load = functools.partial(load_variables, sess=sess)

与

在第4行， 替换

与

这将用save_state和load_state对替换save_变量和loas_变量对

---

希望这能对您有所帮助。

我遇到了同样的问题，我通过对基线代码进行一些调整，解决了这个问题

基线中有两对用于保存和加载模型的方法（save_state&load_state对和save_variables&loas_variables对），您可以在baselines/common/tf_util.py（line325~line372）中看到

对于最新版本的基线，以.ckpt.meta、.ckpt.index、.ckpt.data和检查点格式保存和加载模型的save_state和load_state对已被放弃，因此您需要重新启用save_state和load_state对

以ppo2为例，在baselines/ppo2/model.py中，进行以下替换： 在第125行中，更换

self.save = functools.partial(save_variables, sess=sess)
self.load = functools.partial(load_variables, sess=sess)

与

在第4行， 替换

与

这将用save_state和load_state对替换save_变量和loas_变量对

希望这对你有帮助

from baselines.common.tf_util import get_session, save_variables, load_variables

from baselines.common.tf_util import get_session, save_state, load_state