search

maximilianigl / DVRL

Deep Variational Reinforcement Learning
Apache License 2.0
134 stars 25 forks source link

How long it takes to train the model? #12

Open BigBearBlacken opened 2 years ago

BigBearBlacken commented 2 years ago

I have spent nearly 48 hours since I train this model using cpu. The command is python ./code/main.py -p with environment.config_file=openaiEnv.yaml in the Readme file. And it is still running now. Could you please tell me how long it took when you train the model.

maximilianigl commented 2 years ago

Those jobs are quite compute intensive, as its running many environments in parallel. I don't remember how long it took to train, but I'd guess on the order of about a day - but on a server with 20+ cores and a GPU.

The program should give you some output, based on which you should be able to estimate how far in training you are and how long it will still take.