Ошибка памяти при использовании keras-rl для обучения подкреплению - PullRequest
Новая
       50

Ошибка памяти при использовании keras-rl для обучения подкреплению

0 голосов
/ 13 июня 2019

Я использую 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

Я также переписываю код самостоятельно (не используйте API-интерфейс keras-rl), однако, похожая ошибка произошла.Поскольку я использую 1080Ti и 48G RAM, я действительно озадачен результатом.В чем причина проблемы?Заранее спасибо.

...