Batch size = 16?
mctigger opened this issue · comments
dreamerv2/dreamerv2/configs.yaml
Line 24 in 912ec5d
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.
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.