dreamer-torch icon indicating copy to clipboard operation
dreamer-torch copied to clipboard

Pytorch version of Dreamer, which follows the original TF v2 codes.

Results 4 dreamer-torch issues
Sort by recently updated
recently updated
newest added

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