dreamerv3-torch
Pytorch implementation of Mastering Diverse Domains through World Models.
Validation is in progress.
Instructions
Get dependencies:
pip install -r requirements.txt
Train the agent:
python3 dreamer.py --configs defaults --logdir $ABSOLUTEPATH_TO_SAVE_LOG
Monitor results:
tensorboard --logdir $ABSOLUTEPATH_TO_SAVE_LOG
Acknowledgments
This code is heavily inspired by the following works:
- danijar's Dreamer-v2 tensorflow implementation: https://github.com/danijar/dreamerv2
- jsikyoon's Dreamer-v2 pytorch implementation: https://github.com/jsikyoon/dreamer-torch
- RajGhugare19's Dreamer-v2 pytorch implementation: https://github.com/RajGhugare19/dreamerv2
- denisyarats's DrQ-v2 original implementation: https://github.com/facebookresearch/drqv2
Languages
Python
98.6%
Dockerfile
0.9%
Shell
0.5%