UNABEL加载PPO模型

Question

您好，我已经从stabel_baselines3上培训了一个PPO模型3，我保存了它，

model.save("model")

但是当我尝试加载它时，我会收到以下错误：

m = PPO.load("model", env=env)
AttributeError                            Traceback (most recent call last)
/tmp/ipykernel_25649/121834194.py in <module>
      2 env = e.MinitaurBulletEnv(render=False)
      3 env.reset()
----> 4 m2 = PPO.load("model", env=env)
      5 for episode in range(1, 6):
      6     obs = env.reset()

~/anaconda3/lib/python3.8/site-packages/stable_baselines3/common/base_class.py in load(cls, path, env, device, custom_objects, **kwargs)
    668             env = cls._wrap_env(env, data["verbose"])
    669             # Check if given env is valid
--> 670             check_for_correct_spaces(env, data["observation_space"], data["action_space"])
    671         else:
    672             # Use stored env, if one exists. If not, continue as is (can be used for predict)

~/anaconda3/lib/python3.8/site-packages/stable_baselines3/common/utils.py in check_for_correct_spaces(env, observation_space, action_space)
    217     :param action_space: Action space to check against
    218     """
--> 219     if observation_space != env.observation_space:
    220         raise ValueError(f"Observation spaces do not match: {observation_space} != {env.observation_space}")
    221     if action_space != env.action_space:

~/anaconda3/lib/python3.8/site-packages/gym/spaces/box.py in __eq__(self, other)
    138 
    139     def __eq__(self, other):
--> 140         return isinstance(other, Box) and (self.shape == other.shape) and np.allclose(self.low, other.low) and np.allclose(self.high, other.high)

AttributeError: 'Box' object has no attribute 'shape'

知道env是Pybullet的盒子Env，

import pybullet_envs.bullet.minitaur_gym_env as e
import gym

env = e.MinitaurBulletEnv(render=False)
env.reset()

其他信息是该模型完美地加载了合作

Answer 1

从您的问题来看，我无法分辨您是否在Google Colab上工作，但是如果您是，我认为加载时绝对应该包括保存模型的整个路径。也许您也需要这样做，即使不是在Colab中。

我的意思是，当您加载模型时，您的代码线可能看起来像这样：

m = PPO.load("./model.zip/", env=env)

我希望这会有所帮助！