1.2 KiB
1.2 KiB
Dreamer-v3 Pytorch
Pytorch implementation of Mastering Diverse Domains through World Models
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
Evaluation Results
work-in-progress
Awesome Environments used for testing:
- Deepmind control suite: https://github.com/deepmind/dm_control
- will be added soon
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