Commit Graph

  • a544949ebe Added example of torchvision image augmentation on LeRobotDataset Marina Barannikov 2024-06-05 10:53:18 +00:00
  • fdf56e7a62 Redesign config Simon Alibert 2024-06-05 09:49:31 +00:00
  • 443b06b412 refactor show_image_transforms Simon Alibert 2024-06-05 09:34:39 +00:00
  • 1eb4bfe2e4 Fix videos_dir documentation (#247) Alexander Soare 2024-06-05 08:25:20 +01:00
  • 57d3d27c78 add simple manual real world gym env example Thomas Wolf 2024-06-05 00:37:46 +02:00
  • 21f222fa1d Add out_dir option to eval (#244) Alexander Soare 2024-06-04 20:01:53 +01:00
  • 0935e49c8a add lerobotdataset in readme Thomas Wolf 2024-06-04 18:45:12 +02:00
  • 33362dbd17 Adding parameter dataloading_s to console logs and wandb for tracking… (#243) amandip7 2024-06-04 21:32:05 +05:30
  • 3c2dd1b881 fix test thomwolf_2024_06_04_nans Thomas Wolf 2024-06-04 15:43:10 +02:00
  • 5e85a2c50b Add reachy2 dataset, policy, env Remi Cadene 2024-06-04 12:31:59 +00:00
  • 22bd1f0669 Updated formatting Marina Barannikov 2024-06-04 12:06:36 +00:00
  • 31e3c82386 Merge remote-tracking branch 'refs/remotes/origin/2024_05_30_add_data_augmentation' into 2024_05_30_add_data_augmentation Marina Barannikov 2024-06-04 12:00:46 +00:00
  • 5eea2542d9 Added visualisations for image augmentation Marina Barannikov 2024-06-04 11:57:45 +00:00
  • 42f9cc9c2a Updated transforms arguments Marina Barannikov 2024-06-04 11:14:54 +00:00
  • 66629a956d Updated config to match transforms Marina Barannikov 2024-06-04 11:09:23 +00:00
  • 7be2c35c0a Merge branch 'huggingface:main' into 2024_05_30_add_data_augmentation Simon Alibert 2024-06-04 12:28:06 +02:00
  • b0d954c6e1 Fix bug in normalize to avoid divide by zero (#239) Ruijie 2024-06-04 06:21:28 -04:00
  • 63a5d0be39 fix nans Thomas Wolf 2024-06-04 11:40:46 +02:00
  • eac660bb9e fix nans thomwolf_2024_06_04_nan_fixes Thomas Wolf 2024-06-04 11:40:46 +02:00
  • 1333560f6b Add dora_aloha_real_act_real and dora_aloha_real_act_real_no_state test artifacts Remi Cadene 2024-05-30 17:56:46 +00:00
  • 92d1aecb40 Rename dora_aloha_real, WIP test_policies Remi Cadene 2024-05-30 17:54:59 +00:00
  • 03d237fe0f small fix Remi Cadene 2024-05-30 13:38:19 +00:00
  • 1c9f447ad0 small fix Remi Cadene 2024-05-30 13:36:34 +00:00
  • 73f1a3932d Add dora-lerobot to pyproject Remi Cadene 2024-05-30 13:35:28 +00:00
  • 97bda08e0f Rename Aloha2 to Aloha Remi Cadene 2024-05-30 13:35:02 +00:00
  • 51dea3f67c fix Remi Cadene 2024-05-30 12:10:44 +00:00
  • 5495d55cc7 Add aloha2_real, Add act_real, Fix vae=false, Add support for no state Remi Cadene 2024-05-30 12:06:57 +00:00
  • 14291171cc Updated default.yaml Marina Barannikov 2024-06-03 17:21:16 +00:00
  • cc4b3bd8e7 Updated default.yaml Marina Barannikov 2024-06-03 17:18:33 +00:00
  • 602ea9844b Add RandomSubsetApply Simon Alibert 2024-06-03 17:15:37 +00:00
  • a56626cf9c Add custom visualize_dataset.py Remi Cadene 2024-06-01 15:45:26 +00:00
  • 44ba4ed566 Fix aloha (WIP: do not train in sim) Remi Cadene 2024-06-03 14:47:06 +00:00
  • bd3111f28b Fix visualize_dataset.py --help (#241) Simon Alibert 2024-06-03 16:35:16 +02:00
  • 9f8415fa83 Added clarification comments Marina Barannikov 2024-06-03 14:18:08 +00:00
  • cf15cba5fc Remove redundant slicing operation in Diffusion Policy (#240) Alexander Soare 2024-06-03 13:04:24 +01:00
  • 205e0c9dde Fix ACT delta_timestamps: first action should starts at 1/fps, not 0 Remi Cadene 2024-06-01 16:01:55 +00:00
  • 5b74205e16 Add custom visualize_dataset.py Remi Cadene 2024-06-01 15:45:26 +00:00
  • 042e193995 Typo in examples\4_train_policy_with_script.md (#235) jganitzer 2024-05-31 19:14:14 +02:00
  • 212a5ab29b Updated implementation on MultiLeRobotDataset Marina Barannikov 2024-05-31 16:28:08 +00:00
  • c4870e5892 Added data augmentation feature to MultiLeRobotDataset Marina Barannikov 2024-05-31 15:42:31 +00:00
  • a9561363a5 Merge remote-tracking branch 'origin/main' into user/aliberts/2024_05_06_add_coverage user/aliberts/2024_05_06_add_coverage Simon Alibert 2024-05-31 17:20:02 +02:00
  • 20a3715469 Merge remote-tracking branch 'origin/main' into 2024_05_30_add_data_augmentation Marina Barannikov 2024-05-31 14:50:31 +00:00
  • 65e46a49e1 Implemented data augmentation with LeRobot class marina.barannikov@huggingface.co 2024-05-31 14:16:38 +00:00
  • d585c73f9f Add real-world support for ACT on Aloha/Aloha2 (#228) Remi 2024-05-31 15:31:02 +02:00
  • 504d2aaf48 add EpisodeAwareSampler (#217) Radek Osmulski 2024-05-31 22:43:47 +10:00
  • 9b5d2fd37d fix aloha conversion changes fix_aloha_conversion_nan Thomas Wolf 2024-05-31 11:31:28 +02:00
  • 83f4f7f7e8 Add precision param to format_big_number (#232) Radek Osmulski 2024-05-31 18:19:01 +10:00
  • 633115d861 Fix chaining in MultiLerobotDataset (#233) Alexander Soare 2024-05-31 09:03:28 +01:00
  • 97ea288084 Add dora_aloha_real_act_real and dora_aloha_real_act_real_no_state test artifacts Remi Cadene 2024-05-30 17:56:46 +00:00
  • 671ad93b6c Rename dora_aloha_real, WIP test_policies Remi Cadene 2024-05-30 17:54:59 +00:00
  • 57fb5fe8a6 Improve documentation on VAE encoder inputs (#215) Alexander Soare 2024-05-30 18:16:44 +01:00
  • 0b51a335bc Add a test for MultiLeRobotDataset making sure it produces all frames. (#230) Alexander Soare 2024-05-30 17:46:25 +01:00
  • 111cd58f8a Add MultiLerobotDataset for training with multiple LeRobotDatasets (#229) Alexander Soare 2024-05-30 16:12:21 +01:00
  • b7b5c3b4ff small fix Remi Cadene 2024-05-30 13:38:19 +00:00
  • 1397036a6b small fix Remi Cadene 2024-05-30 13:36:34 +00:00
  • c1570e40c6 Add dora-lerobot to pyproject Remi Cadene 2024-05-30 13:35:28 +00:00
  • 8d847a58ef Rename Aloha2 to Aloha Remi Cadene 2024-05-30 13:35:02 +00:00
  • 48f974bb9e fix Remi Cadene 2024-05-30 12:10:44 +00:00
  • 511e39bdb8 Add aloha2_real, Add act_real, Fix vae=false, Add support for no state Remi Cadene 2024-05-30 12:06:57 +00:00
  • 265b0ec44d Refactor env to add key word arguments from config yaml (#223) Remi 2024-05-30 13:45:22 +02:00
  • 2c2e4e14ed Add aloha_dora_format.py (#201) Remi 2024-05-30 11:26:39 +02:00
  • fbb04b907c Merge remote-tracking branch 'origin/main' into user/aliberts/2024_05_06_add_coverage Simon Alibert 2024-05-30 09:31:42 +02:00
  • 25801e7520 fix attempt Simon Alibert 2024-05-30 09:30:42 +02:00
  • 13310681b1 Enable cuda for end-to-end tests (#222) Simon Alibert 2024-05-29 23:02:23 +02:00
  • 255dbef76a Add offline inference to visualize_dataset user/rcadene/2024_05_20_dora_parquet_format_viz Remi Cadene 2024-05-29 15:33:52 +00:00
  • 31a25d1dba Use training.eval_freq=-1 for real-world data training + Add diffusion_real.yaml Remi Cadene 2024-05-29 15:30:43 +00:00
  • 31b57de633 Refactor make_env Remi Cadene 2024-05-29 15:26:56 +00:00
  • 2ced211864 Move gym_dora to dora-lerobot github Remi Cadene 2024-05-29 15:20:44 +00:00
  • b670d3c43e Update lerobot/scripts/push_dataset_to_hub.py thom_arm Thomas Wolf 2024-05-29 15:30:39 +02:00
  • efd3357124 Update lerobot/scripts/push_dataset_to_hub.py Thomas Wolf 2024-05-29 15:30:33 +02:00
  • 785db44bd5 Update lerobot/common/datasets/push_dataset_to_hub/aloha_dora_format.py Thomas Wolf 2024-05-29 15:30:21 +02:00
  • e2f690e779 proposal for a more general Dora env Thomas Wolf 2024-05-29 15:29:41 +02:00
  • 1a8e5c1c0b remove warning for ignored logged types (WIP) Remi Cadene 2024-05-29 11:09:44 +00:00
  • 63e61385fc Add num_workers and batch_size to default.yaml Remi Cadene 2024-05-29 11:09:16 +00:00
  • 8c1dd0e263 Add observation queue to ACT + refactor into _queues Remi Cadene 2024-05-29 10:21:50 +00:00
  • b24f2d0028 fix modeling act Remi Cadene 2024-05-29 10:04:27 +00:00
  • ad0033ae12 fix diffusion Remi Cadene 2024-05-29 09:50:44 +00:00
  • 68a680a9eb make aloha dora more flexible for A koch arm Thomas Wolf 2024-05-29 11:40:02 +02:00
  • ce5329cf44 push_to_hub less hardcoded Thomas Wolf 2024-05-29 11:39:25 +02:00
  • b47c07fbeb cherry pick me Alexander Soare 2024-05-28 21:37:46 +01:00
  • 220b32441d Act works with n_obs > 1 Remi Cadene 2024-05-28 20:55:02 +00:00
  • 960589849f Fix aloha_dora_format Remi Cadene 2024-05-28 20:37:12 +00:00
  • 1a6602e77c add support for failed episode haixuantao 2024-05-28 21:20:43 +02:00
  • 743812e471 fix timestamps not aligned (TODO: add some tests) Remi Cadene 2024-05-28 19:12:08 +00:00
  • 3d625ae6d3 Handle crop_shape=None in Diffusion Policy (#219) Alexander Soare 2024-05-28 18:27:33 +01:00
  • e3b9f1c19b Add resume training (#205) Alexander Soare 2024-05-28 12:04:23 +01:00
  • 97cb7a2362 save thom-act Thomas Wolf 2024-05-28 11:08:55 +02:00
  • 7ec76ee235 Fix nightly builds (#216) Simon Alibert 2024-05-28 10:43:34 +02:00
  • c2fd06f765 WIP: Add no_state option + fix use_vae=False to ACT Remi Cadene 2024-05-27 13:20:32 +00:00
  • 946d191919 Merge remote-tracking branch 'upstream/main' into add_drop_last_keyframes alexander-soare/add_drop_last_keyframes Alexander Soare 2024-05-27 09:21:44 +01:00
  • 3b86050ab0 throw an error if config.do_maks_loss and action_is_pad not provided in batch (#213) Radek Osmulski 2024-05-27 18:06:26 +10:00
  • 3189ac26b8 fix symlink Remi Cadene 2024-05-22 08:55:42 +00:00
  • 5cb19d589f WIP Remi Cadene 2024-05-17 02:05:44 +00:00
  • 495da6deaf save_freq=10000 Remi Cadene 2024-05-24 17:32:45 +00:00
  • 6d39b73399 Adds a tutorial section on how to use arbitrary configuration files (#206) Alexander Soare 2024-05-24 12:39:11 +01:00
  • f33a06070e WIP Remi Cadene 2024-05-24 09:42:02 +00:00
  • f409bee6b1 WIP Remi Cadene 2024-05-24 09:37:11 +00:00
  • c91ececc75 WIP Remi Cadene 2024-05-24 09:06:29 +00:00
  • 8a10e8442b Add boilerplate code Simon Alibert 2024-05-20 19:16:37 +02:00
  • fc07f0e2bc Merge remote-tracking branch 'origin/main' into user/aliberts/2024_05_06_add_coverage Simon Alibert 2024-05-23 14:39:41 +02:00