34 lines
1.2 KiB
Markdown
34 lines
1.2 KiB
Markdown
|
# Dreamer-v3 Pytorch
|
||
|
|
Pytorch implementation of [Mastering Diverse Domains through World Models](https://arxiv.org/abs/2301.04104v1)
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
## 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
|