dreamer-torch
dreamer-torch copied to clipboard
Pytorch version of Dreamer, which follows the original TF v2 codes.
there is a little bug when setting `config.precision=16` https://github.com/jsikyoon/dreamer-torch/blob/7c2331acd4fa6196d140943e977f23fb177398b3/dreamer.py#L194 which forgets parsing `config.precision`, resulting still using float precision collecting data. I think it should be like below. ``` python env...
When I was using your code, I only founded the Hyperparameters of Walker-Walk but Humanoid-Walk. In the official paper, it has the results of Humanoid-Walk Environment. So could you please...
Hi, While I'm running the code, I found that pytorch JIT (for fast running) is not used in the code. I'm thinking of applying this feature for faster training. Have...
Hi, Great job porting dreamer2 to torch while matching the tf code. I think the Plan2Explore has to be ported over. https://github.com/jsikyoon/dreamer-torch/blob/e42d504ea362ad605c7bbe2a0d89df1d2b3e07f2/exploration.py#L60