yusukeurakami / dreamer-pytorch

pytorch-implementation of Dreamer (Model-based Image RL Algorithm)
MIT License
162 stars 34 forks source link

Implementation time and device? #7

Open 108618035guotingliao opened 3 years ago

108618035guotingliao commented 3 years ago

Hi,

Thanks you for your sharing~ I've implemented it for a while. I have some questions about the time because I spend lots of time on reaching 500K steps. However, the original authors pointed out that they only spent 3 hrs per 100K steps with Nvidia V100. And I spent about 2 days per 100K steps with RTX 3080. I doubt if my GPU ability was't be properly installed. My GPU's memory is full. But power only reach about 200 Watt, 340 Watt in total.

So, I want to check that how much time you spent and which device you trained? The following is my device.

Thanks you!!!