carpedm20 / deep-rl-tensorflow

TensorFlow implementation of Deep Reinforcement Learning papers

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

MemoryError when running the examples

ethene opened this issue · comments

I'm getting the MemoryError. Ubuntu /2Gb ram + 4 Gb gwap:

Traceback (most recent call last):
File "main.py", line 168, in
tf.app.run()
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/platform/app.py", line 43, in run
sys.exit(main(sys.argv[:1] + flags_passthrough))
File "main.py", line 160, in main
agent = TrainAgent(sess, pred_network, env, stat, conf, target_network=target_network)
File "/home/strky/deep-rl-tensorflow/agents/deep_q.py", line 13, in init
super(DeepQ, self).init(sess, pred_network, env, stat, conf, target_network=target_network)
File "/home/strky/deep-rl-tensorflow/agents/agent.py", line 53, in init
conf.batch_size, conf.history_length, conf.memory_size, conf.observation_dims)
File "/home/strky/deep-rl-tensorflow/agents/experience.py", line 15, in init
self.observations = np.empty([self.memory_size] + observation_dims, dtype=np.uint8)
MemoryError

You memory (even with swap) is not enough to hold the experience replay buffer.