hongshaorou/dreamerv3-torch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

modified based on author's implementation

Browse Source
This commit is contained in:
NM512
2023-03-18 08:38:23 +09:00
parent a678a509b9
commit 6273444394
6 changed files with 371 additions and 229 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
README.md
View File

@@ -17,8 +17,17 @@ Monitor results:
tensorboard --logdir $ABSOLUTEPATH_TO_SAVE_LOG
```
## ToDo
- [ ] Prototyping
- [ ] Modify implementation details based on the author's implementation
- [ ] Evaluate on visual DMC suite(~10 tasks)
- [ ] Add other tasks and corresponding model sizes implementation
- [ ] Continuous implementation improvement
## Acknowledgments
This code is heavily inspired by the following works:
- danijar's Dreamer-v3 jax implementation: https://github.com/danijar/dreamerv3
- 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