Commit Graph

477 Commits

Author SHA1 Message Date
Remi 5c376c14ca
Update examples/1_load_hugging_face_dataset.py
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
2024-04-22 11:33:49 +02:00
Simon Alibert c0cc53cb54
Fixes for datasets 2.18 -> 2.19 update (#88) 2024-04-20 15:06:11 +02:00
Cadene 2a59825a00 fix online training 2024-04-20 00:12:34 +00:00
Cadene 06628ba059 fix online training 2024-04-19 23:58:38 +00:00
Cadene b2b5329683 fix online training 2024-04-19 23:48:43 +00:00
Cadene 85f1554da8 fix visualize_dataset 2024-04-19 23:40:35 +00:00
Cadene 9b4c2e2a9f small fix 2024-04-19 23:30:39 +00:00
Cadene 20928021c0 Add tests/data 2024-04-19 23:27:11 +00:00
Cadene c20cf2fbbc Remove Prod, Tests are passind 2024-04-19 23:27:10 +00:00
Cadene 35a573c98e Use v1.1, hf_transform_to_torch, Add 3 xarm datasets 2024-04-19 23:26:13 +00:00
Cadene 714a776277 id -> index, finish moving compute_stats before hf_dataset push_to_hub 2024-04-19 23:25:06 +00:00
Cadene 64b09ea7a7 WIP add load functions + episode_data_index 2024-04-19 23:24:08 +00:00
Cadene 0bd2ca8d82 Add meta_data, revision v1.1 2024-04-19 23:24:08 +00: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