search

carpedm20 / deep-rl-tensorflow

TensorFlow implementation of Deep Reinforcement Learning papers
MIT License
1.6k stars 396 forks source link

MemoryError when running the examples #20

Closed ethene closed 7 years ago

ethene commented 7 years ago

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

ppwwyyxx commented 7 years ago

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