MouseHu / GEM

13 stars 4 forks source link

Dear authors, How many hours did you take to run each environment? #1

Closed GoingMyWay closed 2 years ago

GoingMyWay commented 2 years ago

Dear authors,

How many hours did you take to run each environment? For example on RTX 3090, how many hours, GPU memories and GPU memories do you need to run each environment?

Best regards.

MouseHu commented 2 years ago

For mujoco tasks, the experiments were run on GTX 1080Ti, and it took roughly 10 hours for 1M steps with 5 seeds running on the same card. Each seed consumes ~500 MB GPU memories, which is the same as TD3.

GoingMyWay commented 2 years ago

Thanks. Did Atari tasks take 10 hours?

HeavyCrab commented 2 years ago

It took roughly 15 hrs for 1M steps with 4 seeds running on the same card. Each seed consumes ~2000MB GPU memories.

GoingMyWay commented 2 years ago

Thank you for your answer. 👍