Alexander Soare
|
abbb1d2367
|
Make sure policies don't mutate the batch (#323)
|
2024-07-22 20:38:33 +01:00 |
|
Simon Alibert
|
0b21210d72
|
Convert datasets to av1 encoding (#302)
|
2024-07-22 20:08:59 +02:00 |
|
Simon Alibert
|
8865e19c12
|
Fix datasets missing versions (#318)
|
2024-07-16 23:02:31 +02:00 |
|
Alexander Soare
|
c0101f0948
|
Fix ACT temporal ensembling (#319)
|
2024-07-16 10:27:21 +01:00 |
|
Remi
|
5e54e39795
|
Add real robot devices and scripts to control real robot (#288)
Co-authored-by: Simon Alibert <alibert.sim@gmail.com>
|
2024-07-15 17:43:10 +02:00 |
|
Remi
|
5ffcb48a9a
|
Add available list of raw repo ids (#312)
|
2024-07-13 11:30:50 +02:00 |
|
Alexander Soare
|
471eab3d7e
|
Make ACT compatible with "observation.environment_state" (#314)
|
2024-07-11 13:12:22 +01:00 |
|
Seungjae Lee
|
64425d5e00
|
Bug fix: fix error when setting select_target_actions_indices in vqbet (#310)
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
|
2024-07-10 17:56:11 +01:00 |
|
Simon Alibert
|
e410e5d711
|
Improve video benchmark (#282)
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
Co-authored-by: Remi <re.cadene@gmail.com>
|
2024-07-09 20:20:25 +02:00 |
|
Alexander Soare
|
cc2f6e7404
|
Train diffusion pusht_keypoints (#307)
Co-authored-by: Remi <re.cadene@gmail.com>
|
2024-07-09 12:35:50 +01:00 |
|
Alexander Soare
|
a4d77b99f0
|
Include observation.environment_state with keypoints in PushT dataset (#303)
Co-authored-by: Remi <re.cadene@gmail.com>
|
2024-07-09 08:27:40 +01:00 |
|
Simon Alibert
|
74362ac453
|
Add VQ-BeT copyrights (#299)
|
2024-07-04 13:02:31 +02:00 |
|
Alexander Soare
|
342f429f1c
|
Add test to make sure policy dataclass configs match yaml configs (#292)
|
2024-06-26 09:09:40 +01:00 |
|
Seungjae Lee
|
7d1542cae1
|
Add VQ-BeT (#166)
|
2024-06-26 08:55:02 +01:00 |
|
Simon Alibert
|
2abef3bef9
|
Enable video_reader backend (#220)
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
|
2024-06-19 17:15:25 +02:00 |
|
Thomas Wolf
|
48951662f2
|
Bug fix: missing attention mask in VAE encoder in ACT policy (#279)
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
|
2024-06-19 12:07:21 +01:00 |
|
Jihoon Oh
|
b72d574891
|
fix Unet global_cond_dim to use state dim, not action dim (#278)
|
2024-06-17 15:17:28 +01:00 |
|
Alexander Soare
|
15dd682714
|
Add multi-image support to diffusion policy (#218)
|
2024-06-17 08:11:20 +01:00 |
|
Thomas Wolf
|
125bd93e29
|
Improve push_dataset_to_hub API + Add unit tests (#231)
Co-authored-by: Remi <re.cadene@gmail.com>
Co-authored-by: Simon Alibert <alibert.sim@gmail.com>
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
|
2024-06-13 15:18:02 +02:00 |
|
Marina Barannikov
|
c38f535c9f
|
FIx make_dataset to match transforms config (#264)
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
|
2024-06-12 19:45:42 +02:00 |
|
Marina Barannikov
|
ff8f6aa6cd
|
Add data augmentation in LeRobotDataset (#234)
Co-authored-by: Simon Alibert <alibert.sim@gmail.com>
Co-authored-by: Remi Cadene <re.cadene@gmail.com>
|
2024-06-11 19:20:55 +02:00 |
|
Wael Karkoub
|
54c9776bde
|
Improves Type Annotations (#252)
|
2024-06-10 19:09:48 +01:00 |
|
Ruijie
|
b0d954c6e1
|
Fix bug in normalize to avoid divide by zero (#239)
Co-authored-by: rj <rj@teleopstrio-razer.lan>
Co-authored-by: Remi <re.cadene@gmail.com>
|
2024-06-04 12:21:28 +02:00 |
|
Alexander Soare
|
cf15cba5fc
|
Remove redundant slicing operation in Diffusion Policy (#240)
|
2024-06-03 13:04:24 +01:00 |
|
Remi
|
d585c73f9f
|
Add real-world support for ACT on Aloha/Aloha2 (#228)
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
|
2024-05-31 15:31:02 +02:00 |
|
Radek Osmulski
|
504d2aaf48
|
add EpisodeAwareSampler (#217)
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
|
2024-05-31 13:43:47 +01:00 |
|
Radek Osmulski
|
83f4f7f7e8
|
Add precision param to format_big_number (#232)
|
2024-05-31 10:19:01 +02:00 |
|
Alexander Soare
|
633115d861
|
Fix chaining in MultiLerobotDataset (#233)
|
2024-05-31 09:03:28 +01:00 |
|
Alexander Soare
|
57fb5fe8a6
|
Improve documentation on VAE encoder inputs (#215)
|
2024-05-30 19:16:44 +02:00 |
|
Alexander Soare
|
111cd58f8a
|
Add MultiLerobotDataset for training with multiple LeRobotDatasets (#229)
|
2024-05-30 16:12:21 +01:00 |
|
Remi
|
265b0ec44d
|
Refactor env to add key word arguments from config yaml (#223)
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
|
2024-05-30 13:45:22 +02:00 |
|
Remi
|
2c2e4e14ed
|
Add aloha_dora_format.py (#201)
Co-authored-by: Thomas Wolf <thomwolf@users.noreply.github.com>
|
2024-05-30 11:26:39 +02:00 |
|
Alexander Soare
|
3d625ae6d3
|
Handle crop_shape=None in Diffusion Policy (#219)
|
2024-05-28 18:27:33 +01:00 |
|
Alexander Soare
|
e3b9f1c19b
|
Add resume training (#205)
Co-authored-by: Remi <re.cadene@gmail.com>
|
2024-05-28 12:04:23 +01:00 |
|
Radek Osmulski
|
3b86050ab0
|
throw an error if config.do_maks_loss and action_is_pad not provided in batch (#213)
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
|
2024-05-27 09:06:26 +01:00 |
|
Remi
|
c4da689171
|
Hot fix to compute validation loss example test (#200)
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
|
2024-05-20 18:30:11 +02:00 |
|
Radek Osmulski
|
9b62c25f6c
|
Adds split_by_episodes to LeRobotDataset (#158)
|
2024-05-20 14:04:04 +02:00 |
|
Remi
|
01eae09ba6
|
Fix aloha real-world datasets (#175)
|
2024-05-20 13:48:09 +02:00 |
|
Alexander Soare
|
5ec0af62c6
|
Explain why n_encoder_layers=1 (#193)
|
2024-05-17 15:05:40 +01:00 |
|
Alexander Soare
|
625f0557ef
|
Act temporal ensembling (#186)
|
2024-05-17 14:57:49 +01:00 |
|
Alexander Soare
|
4d7d41cdee
|
Fix act action queue (#185)
|
2024-05-16 15:43:25 +01:00 |
|
Akshay Kashyap
|
c9069df9f1
|
Port SpatialSoftmax and remove Robomimic dependency (#182)
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
|
2024-05-16 15:34:10 +01:00 |
|
Alexander Soare
|
68c1b13406
|
Make policies compatible with other/multiple image keys (#149)
|
2024-05-16 13:51:53 +01:00 |
|
Simon Alibert
|
f52f4f2cd2
|
Add copyrights (#157)
|
2024-05-15 12:13:09 +02:00 |
|
AshisGhosh
|
fc5cf3d84a
|
Fixes issue #152 - error with creating wandb artifact (#172)
Co-authored-by: Ashis Ghosh <ahsisghosh@live.com>
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
|
2024-05-12 08:13:12 +02:00 |
|
Alexander Soare
|
1249aee3ac
|
Enable logging all the information returned by the forward methods of policies (#151)
|
2024-05-10 07:45:32 +01:00 |
|
Alexander Soare
|
b187942db4
|
Add context manager for seeding (#164)
|
2024-05-09 17:58:39 +01:00 |
|
Ikko Eltociear Ashimine
|
0ea7a8b2a3
|
refactor: update configuration_tdmpc.py (#153)
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
|
2024-05-08 18:13:51 +01:00 |
|
Akshay Kashyap
|
460df2ccea
|
Support for DDIMScheduler in Diffusion Policy (#146)
|
2024-05-08 18:05:16 +01:00 |
|
Alexander Soare
|
f5de57b385
|
Fix SpatialSoftmax input shape (#150)
|
2024-05-08 14:57:29 +01:00 |
|