gsurma / atari

AI research environment for the Atari 2600 games 🤖.
https://gsurma.github.io
MIT License
255 stars 81 forks source link

Training took too much time #17

Open Kingward1017 opened 5 months ago

Kingward1017 commented 5 months ago

Hi!@gsurma Thank you for sharing code I have some problem while running this programme. My GPU device is RTX 3060 laptop , which should be enough to finish this training in time . But I only got this figure after 40h . It's obvious that the episode is not enough , 10 time less than your example. My device cost around 15ms per step . From this and your markdown , I roughly calculated that 40h is enough. What can I do to solve this? loss