Seungjae Lee
|
7d1542cae1
|
Add VQ-BeT (#166)
|
2024-06-26 08:55:02 +01:00 |
Alexander Soare
|
9aa4cdb976
|
Checkpoint on final step of training even when it doesn't coincide with `save_freq`. (#284)
|
2024-06-20 08:27:01 +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 |
|
Thomas Wolf
|
56199fb76f
|
Update readme to detail the lerobot dataset format (#275)
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
|
2024-06-18 13:40:03 +01:00 |
|
Thomas Wolf
|
11f1cb5dc9
|
Bug fix: fix setting different learning rates between backbone and main model in ACT policy (#280)
|
2024-06-18 13:31:35 +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 |
Marina Barannikov
|
e28fa2344c
|
added visualization for min and max transforms (#271)
Co-authored-by: Simon Alibert <alibert.sim@gmail.com>
|
2024-06-17 09:09:57 +02:00 |
|
Simon Alibert
|
a92d79fff2
|
Fix nightlies (#273)
|
2024-06-14 17:11:19 +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 |
Ikko Eltociear Ashimine
|
1cf050d412
|
chore: update 4_train_policy_with_script.md (#257)
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
|
2024-06-11 08:24:39 +01:00 |
Wael Karkoub
|
54c9776bde
|
Improves Type Annotations (#252)
|
2024-06-10 19:09:48 +01:00 |
Luc Georges
|
a06598678c
|
feat(ci): add trufflehog secrets detection (#254)
|
2024-06-10 14:25:43 +02:00 |
Thomas Lips
|
055a6f60c6
|
add root argument to the dataset visualizer to visualize local datasets (#249)
|
2024-06-10 10:44:32 +02:00 |
|
Simon Alibert
|
e54d6ea1eb
|
Make `display_sys_info.py` install-agnostic (#253)
|
2024-06-07 15:02:17 +02:00 |
|
Alexander Soare
|
1eb4bfe2e4
|
Fix videos_dir documentation (#247)
|
2024-06-05 08:25:20 +01:00 |
|
Alexander Soare
|
21f222fa1d
|
Add out_dir option to eval (#244)
|
2024-06-04 21:01:53 +02:00 |
amandip7
|
33362dbd17
|
Adding parameter dataloading_s to console logs and wandb for tracking… (#243)
Co-authored-by: Remi <re.cadene@gmail.com>
|
2024-06-04 17:02:05 +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 |
|
Simon Alibert
|
bd3111f28b
|
Fix `visualize_dataset.py --help` (#241)
|
2024-06-03 16:35:16 +02:00 |
|
Alexander Soare
|
cf15cba5fc
|
Remove redundant slicing operation in Diffusion Policy (#240)
|
2024-06-03 13:04:24 +01:00 |
jganitzer
|
042e193995
|
Typo in examples\4_train_policy_with_script.md (#235)
|
2024-05-31 18:14:14 +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
|
0b51a335bc
|
Add a test for MultiLeRobotDataset making sure it produces all frames. (#230)
Co-authored-by: Remi <re.cadene@gmail.com>
|
2024-05-30 17:46:25 +01:00 |
|
Alexander Soare
|
111cd58f8a
|
Add `MultiLerobotDataset` for training with multiple `LeRobotDataset`s (#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 |
|
Simon Alibert
|
13310681b1
|
Enable cuda for end-to-end tests (#222)
|
2024-05-29 23:02:23 +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 |
|
Simon Alibert
|
7ec76ee235
|
Fix nightly builds (#216)
|
2024-05-28 10:43:34 +02: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 |
|
Alexander Soare
|
6d39b73399
|
Adds a tutorial section on how to use arbitrary configuration files (#206)
|
2024-05-24 12:39:11 +01:00 |
|
Simon Alibert
|
aca424a481
|
Add dev docker image (#189)
Co-authored-by: Remi <re.cadene@gmail.com>
|
2024-05-23 14:39:14 +02:00 |
|
Simon Alibert
|
35c1ce7a66
|
Fix install issues (#191)
|
2024-05-23 14:25:18 +02:00 |
|
Alexander Soare
|
e67da1d7a6
|
Add tutorials for using the training script and (#196)
Co-authored-by: Remi <re.cadene@gmail.com>
|
2024-05-21 16:47:49 +01:00 |
|
Alexander Soare
|
b6c216b590
|
Add Automatic Mixed Precision option for training and evaluation. (#199)
|
2024-05-20 18:57:54 +01:00 |
|
Alexander Soare
|
2b270d085b
|
Disable online training (#202)
Co-authored-by: Remi <re.cadene@gmail.com>
|
2024-05-20 18:27:54 +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
|
19dfb9144a
|
Update the README to reflect WandB disabled by default (#198)
|
2024-05-20 09:02:24 +01:00 |
|
Alexander Soare
|
096149b118
|
Disable wandb by default (#195)
|
2024-05-17 18:01:39 +01:00 |