I would like to ask how much memory is needed to run the code?

inoryy / reaver

Reaver: Modular Deep Reinforcement Learning Framework. Focused on StarCraft II. Supports Gym, Atari, and MuJoCo.

MIT License

554 stars 89 forks source link

I would like to ask how much memory is needed to run the code? #5

Closed mushroom1116 closed 6 years ago

mushroom1116 commented 6 years ago

I received some errors when running the code. OOM when allocating tensor with shape[512,1850,32,32] So I would like to ask how much memory is needed to run this code? Thank you.

inoryy commented 6 years ago

Out of the box the project requires around 10.5 GB of GPU memory to run. If that doesn't fit, you have two options:

reduce resolution of screen/minimap (16px games are still reasonable)
remove some of the features, which can be configured with a json file (example)