Dorsa Rohani
|
c10f875ff0
|
Add Diffusion Policy for Reinforcement Learning (#9824)
* enable cpu ability
* model creation + comprehensive testing
* training + tests
* all tests working
* remove unneeded files + clarify docs
* update train tests
* update readme.md
* remove data from gitignore
* undo cpu enabled option
* Update README.md
* update readme
* code quality fixes
* diffusion policy example
* update readme
* add pretrained model weights + doc
* add comment
* add documentation
* add docstrings
* update comments
* update readme
* fix code quality
* Update examples/reinforcement_learning/README.md
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
* Update examples/reinforcement_learning/diffusion_policy.py
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
* suggestions + safe globals for weights_only=True
* suggestions + safe weights loading
* fix code quality
* reformat file
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
|
2024-11-02 09:18:44 +05:30 |
|