Add Diffusion Policy for Reinforcement Learning #9824
sayakpaul
merged
34 commits into
huggingface:main
from
DorsaRoh:diffusion-policy
enable cpu ability
654f6b4a
model creation + comprehensive testing
96c62d09
training + tests
8759c122
all tests working
2cbc3bb9
remove unneeded files + clarify docs
5feb7af5
Merge branch 'diffusion-policy' of https://github.com/DorsaRoh/diffus…
80fcab8a
update train tests
a4c73408
update readme.md
6457aec9
remove data from gitignore
ddb77186
undo cpu enabled option
089c40ac
Update README.md
7d962540
update readme
25e4638d
Merge branch 'diffusion-policy' of https://github.com/DorsaRoh/diffus…
566f112d
Merge branch 'main' into diffusion-policy
fbb442c0
code quality fixes
4c07ceaa
Merge branch 'diffusion-policy' of https://github.com/DorsaRoh/diffus…
c0e1af3d
diffusion policy example
957de92c
update readme
09922dfc
sayakpaul
commented
on 2024-11-01
add pretrained model weights + doc
186b8f06
add comment
d7ced53e
Merge branch 'main' into diffusion-policy
333b5cb2
add documentation
b37b0e22
Merge branch 'diffusion-policy' of https://github.com/DorsaRoh/diffus…
65dd0b71
add docstrings
06401cb8
update comments
2bf4fa0d
update readme
47f2ba50
fix code quality
be98e767
Update examples/reinforcement_learning/README.md
5bff0397
Update examples/reinforcement_learning/diffusion_policy.py
2bad2fc2
suggestions + safe globals for weights_only=True
a7b8ef28
Merge branch 'diffusion-policy' of https://github.com/DorsaRoh/diffus…
9a5cfe72
suggestions + safe weights loading
19db70eb
fix code quality
c8fa61a6
reformat file
b8fe110a
sayakpaul
commented
on 2024-11-02
sayakpaul
approved these changes
on 2024-11-02
Login to write a write a comment.
Login via GitHub