Commit Graph

742 Commits

Author SHA1 Message Date
Cadene b241ea46dd move download_and_upload_dataset.py to root_dir 2024-04-16 17:20:53 +00:00
Cadene e09d25267e fix online training 2024-04-16 17:20:53 +00:00
Cadene 4a3eac4743 fix unit tests, stats was missing, visualize_dataset was broken 2024-04-16 17:20:53 +00:00
Cadene 69eeced9d9 add datasets to poetry/cpu 2024-04-16 17:20:52 +00:00
Cadene 0980fff6cc HF datasets works 2024-04-16 17:19:40 +00:00
Cadene 5edd9a89a0 Move stats_dataset init into else statement -> faster init 2024-04-16 17:19:39 +00:00
Cadene c7a8218620 typo 2024-04-16 17:19:39 +00:00
Cadene 67d79732f9 Add download_and_upload_dataset.py in script, update all datasets, update online training 2024-04-16 17:19:39 +00:00
Cadene c6aca7fe44 For Pusht: use hf datasets to train, rename load_data_with_delta_timestamps -> load_previous_and_future_frames 2024-04-16 17:19:06 +00:00
Alexander Soare cb3978b5f3 backup wip 2024-04-16 18:12:39 +01:00
Remi 4ed55c3ba3
Merge pull request #76 from huggingface/user/aliberts/2024_04_16_refactor_pyproject
Refactor pyproject
2024-04-16 18:55:09 +02:00
Alexander Soare 0eb899de73 Merge remote-tracking branch 'upstream/main' into unify_policy_api 2024-04-16 17:30:41 +01:00
Alexander Soare a8ddefcfd4
Merge pull request #71 from huggingface/user/alexander-soare/refactor_dp
Partial refactor of Diffusion Policy
2024-04-16 17:24:05 +01:00
Alexander Soare a9496fde39 revision 1 2024-04-16 17:15:51 +01:00
Alexander Soare 8a322da422 backup wip 2024-04-16 16:35:04 +01:00
Alexander Soare 23be5e1e7b backup wip 2024-04-16 16:31:44 +01:00
Alexander Soare 43a614c173 Fix test_examples 2024-04-16 14:07:16 +01:00
Alexander Soare 9c2f10bd04 ready for review 2024-04-16 13:43:58 +01:00
Alexander Soare 03b08eb74e backup wip 2024-04-16 12:51:32 +01:00
Simon Alibert 38ef878eed Add COC badge 2024-04-16 13:37:20 +02:00
Simon Alibert dee174f678 pre-commit autoupdate 2024-04-16 12:10:26 +02:00
Simon Alibert 70e8de95f7 Clean & update pyproject 2024-04-16 12:09:56 +02:00
Simon Alibert 376d75f8d3 Add env info 2024-04-16 10:35:43 +02:00
Simon Alibert a621ec8d88 Add PR & issue templates 2024-04-16 10:34:29 +02:00
Simon Alibert f9b48abe64 Add CONTRIBUTING 2024-04-16 10:33:58 +02:00
Simon Alibert 352010b718 Add COC 2024-04-16 10:32:09 +02:00
Alexander Soare 5608e659e6 backup wip 2024-04-15 19:06:44 +01:00
Alexander Soare 14f3ffb412 Merge remote-tracking branch 'upstream/main' into refactor_dp 2024-04-15 17:08:28 +01:00
Alexander Soare e37f4e8c53
Merge pull request #72 from huggingface/alexander-soare/policy_config
Use policy configs instead of passing arguments directly to policy classes
2024-04-15 16:44:21 +01:00
Alexander Soare 30023535f9 revision 1 2024-04-15 10:56:43 +01:00
Alexander Soare 40d417ef60 Make sure to make remove all traces of omegaconf from policy config 2024-04-15 09:59:18 +01:00
Alexander Soare 9241b5e830 pass step as kwarg 2024-04-15 09:52:54 +01:00
Alexander Soare 2ccf89d78c try fix tests 2024-04-15 09:48:03 +01:00
Alexander Soare ef4bd9e25c Use dataclass config for ACT 2024-04-15 09:39:23 +01:00
Alexander Soare 34f00753eb remove policy.py 2024-04-12 17:13:25 +01:00
Alexander Soare 55e484124a draft pr 2024-04-12 17:03:59 +01:00
Alexander Soare 6d0a45a97d ready for review 2024-04-12 11:36:52 +01:00
Alexander Soare 5666ec3ec7 backup wip 2024-04-11 18:33:54 +01:00
Alexander Soare 94cc22da9e Merge remote-tracking branch 'upstream/main' into refactor_dp 2024-04-11 17:52:10 +01:00
Alexander Soare 976a197f98 backup wip 2024-04-11 17:51:35 +01:00
Remi 5bd953e8e7
Merge pull request #64 from huggingface/user/rcadene/2024_03_31_remove_torchrl
Remove torchrl
2024-04-11 16:45:16 +02:00
Cadene c1a618e567 fix pusht images type from float32 to uint8, update gym-pusht dependencies 2024-04-11 14:29:16 +00:00
Cadene 4216636084 fix 2024-04-11 14:01:23 +00:00
Cadene 8e5b4365ac fix 2024-04-11 13:57:22 +00:00
Cadene 84d2468da1 fix 2024-04-11 13:51:13 +00:00
Cadene 5be83fbff6 fix 2024-04-11 13:46:13 +00:00
Cadene a605eec7e9 Use pusht as example so it's fast 2024-04-11 13:36:50 +00:00
Cadene 2f4af32d3f small fix 2024-04-11 13:21:06 +00:00
Cadene 36de77ac18 fix and clarify tests 2024-04-11 13:16:47 +00:00
Cadene 92701088a3 small fix 2024-04-11 13:04:27 +00:00