Commit Graph

467 Commits

Author SHA1 Message Date
Remi e760e4cd63
Move normalization to policy for act and diffusion (#90)
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
2024-04-25 11:47:38 +02:00
Remi c1bcf857c5
Fix online training (#94) 2024-04-23 18:54:55 +02:00
Remi 1030ea0070
Loads episode_data_index and stats during dataset __init__ (#85)
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
2024-04-23 14:13:25 +02:00
Alexander Soare e2168163cd
Quality of life patches for eval.py (#86) 2024-04-19 12:33:47 +01:00
Simon Alibert ac0ab27333
Hotfix test_examples.py (#87) 2024-04-19 12:36:04 +02:00
Alexander Soare 8d980940a2
Fix tolerance for delta_timestamps (#84)
Co-authored-by: Remi <re.cadene@gmail.com>
2024-04-18 18:48:22 +01:00
Simon Alibert 7ad1909641
Tests cleaning & simplification (#81) 2024-04-18 14:47:42 +02:00
Remi 0928afd37d
Improve dataset examples (#82)
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
2024-04-18 11:43:16 +02:00
Alexander Soare d5c4b0c344
Merge pull request #80 from huggingface/alexander-soare/unify_policy_api
Unify policy API
2024-04-17 17:05:22 +01:00
Alexander Soare dd9c6eed15 Add temporary patch in TD-MPC 2024-04-17 16:27:57 +01:00
Alexander Soare 2298ddf226 wip 2024-04-17 16:21:37 +01:00
Alexander Soare 63e5ec6483 revert some formatting changes 2024-04-17 11:40:49 +01:00
Alexander Soare c50a13ab31 draft 2024-04-17 10:50:54 +01:00
Alexander Soare 18dd8f32cd Merge remote-tracking branch 'upstream/main' into unify_policy_api 2024-04-17 09:10:30 +01:00
Alexander Soare 6d78bca749
Merge pull request #77 from huggingface/alexander-soare/fix_save_stats
Fix issue with saving freshly computed stats
2024-04-17 09:09:18 +01:00
Alexander Soare 296bbfe1ad Merge branch 'fix_stats_saving' into unify_policy_api 2024-04-17 09:08:04 +01:00
Alexander Soare c9454333d8 revision 2024-04-17 09:02:35 +01:00
Alexander Soare 1331d3b4e4 fix issue with saving freshly computed stats 2024-04-17 08:49:28 +01:00
Alexander Soare bff4b673c9 Merge remote-tracking branch 'upstream/main' into unify_policy_api 2024-04-17 08:08:57 +01:00
Remi 3f1c322d56
Merge pull request #73 from huggingface/user/rcadene/2024_04_14_hf_datasets
Use Hugging Face datasets.Dataset
2024-04-16 21:54:37 +02:00
Simon Alibert fbc31d906c
Merge pull request #74 from huggingface/user/aliberts/2024_04_15_setup_contributions
Setup contributions
2024-04-16 19:51:13 +02:00
Simon Alibert 4057cc6c28
Apply suggestions from code review
Various fixes for #74

Co-authored-by: Remi <re.cadene@gmail.com>
2024-04-16 19:35:01 +02:00
Cadene 91badebdfc fix tests 2024-04-16 17:29:31 +00:00
Cadene 4327e43f19 fix merge thingy 2024-04-16 17:24:25 +00:00
Cadene 36d9e885ef Address comments 2024-04-16 17:20:54 +00:00
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