quanvuong / handful-of-trials-pytorch

Unofficial Pytorch code for "Deep Reinforcement Learning in a Handful of Trials using Probabilistic Dynamics Models"
186 stars 34 forks source link

Where is the neural network for the dynamics defined? #1

Closed dingchenghu closed 5 years ago

dingchenghu commented 5 years ago

Thanks

quanvuong commented 5 years ago

Hi,

It's defined in the config folder. The NN for each environment is defined in the file with the same name as the environment's name.

https://github.com/quanvuong/handful-of-trials-pytorch/tree/master/config

dingchenghu commented 5 years ago

Oh sorry I didn't see that. Got it! Thank you so much