0

我使用keras-rl并成功运行了keras-rl的示例，即dqn_cartpole.py。然后我更改 env_name 来玩 Pong 游戏，即 env_name = "PongNoFrameskip-v4"。一切看起来都很好，但是程序突然出现内存错误，如下所示：

Traceback (most recent call last):
  File "D:/github_repos/keras-rl-master/examples/dqn_cartpole.py", line 47, in <module>
    dqn.fit(env, nb_steps=500000, visualize=True, verbose=2)
  File "D:\github_repos\keras-rl-master\rl\core.py", line 224, in fit
    callbacks.on_episode_end(episode, episode_logs)
  File "D:\github_repos\keras-rl-master\rl\callbacks.py", line 67, in on_episode_end
    callback.on_episode_end(episode, logs=logs)
  File "D:\github_repos\keras-rl-master\rl\callbacks.py", line 191, in on_episode_end
    'obs_mean': np.mean(self.observations[episode]),
  File "C:\Anaconda3\lib\site-packages\numpy\core\fromnumeric.py", line 2920, in mean
    out=out, **kwargs)
  File "C:\Anaconda3\lib\site-packages\numpy\core\_methods.py", line 59, in _mean
    arr = asanyarray(a)
  File "C:\Anaconda3\lib\site-packages\numpy\core\numeric.py", line 553, in asanyarray
    return array(a, dtype, copy=False, order=order, subok=True)
MemoryError

我也自己重写了代码（不要使用keras-rl的api），但是出现了类似的错误。当我使用 1080Ti 和 48G 内存时，我对结果感到非常困惑。什么原因导致问题？提前致谢。

4

0 回答 0