search

danijar / dreamerv2

Mastering Atari with Discrete World Models
https://danijar.com/dreamerv2
MIT License
886 stars 195 forks source link

Batch size = 16? #35

Closed mctigger closed 2 years ago

mctigger commented 2 years ago

https://github.com/danijar/dreamerv2/blob/912ec5da79467b22917cce683c776f034850f91d/dreamerv2/configs.yaml#L24

Hi Danijar, do I understand correct that this line should have batch = 50 to to have same hyperparameters as in the paper? I am asking because I want to investigate why my own PyTorch implementation is slower.

danijar commented 2 years ago

Yep, the defaults in the repo here are tuned for getting quick results (and top performance is still high). The hparams in the paper are a bit different.