search

hzwer / ICCV2019-LearningToPaint

ICCV2019 - Learning to Paint With Model-based Deep Reinforcement Learning
MIT License
2.25k stars 312 forks source link

Training parameters #43

Closed patrisaru closed 3 years ago

patrisaru commented 3 years ago

Hi !

I am trying to train the paint agent in my GPU. In the paper I could read that the training time was about 2 days in your case.

Can you tell me what parameters used you to train the paint agent? In my case the training time is more than 1 week (I am training the agent in a GPU too but I think that there is a lot of time difference).

Thanks so much!

hzwer commented 3 years ago

Hi, I use the default parameter and train the agent on a 2080Ti GPU. I think you can try python-CProfile to check whether there is a bottleneck in the data generator.