Commit Graph

667 Commits

Author SHA1 Message Date
Remi Cadene d525e1b0f8 Add keyboard interaction, Add tqdm, Optimize stuff, Fix, Add resuming 2024-07-12 19:56:28 +02:00
Remi Cadene 7a659dbd6b Add reset-time-s, Add keyboard early exit, Add comments 2024-07-12 12:58:56 +02:00
Remi Cadene 1993d29296 Remove int32 convertion function, Set PID 1500, 0, 400 for elbow 2024-07-12 11:24:29 +02:00
Remi Cadene e615176845 nit 2024-07-11 20:03:56 +02:00
Remi Cadene 3918e118c3 Add assert_same_address 2024-07-11 18:38:23 +02:00
Remi Cadene 17bc32c1e3 nit 2024-07-11 17:16:24 +02:00
Remi Cadene 73692e7459 Make data recording smooth and small fix 2024-07-11 17:14:23 +02:00
Remi Cadene e6aa8cc641 more ports 2024-07-11 15:27:03 +02:00
Alexander Soare 471eab3d7e
Make ACT compatible with "observation.environment_state" (#314) 2024-07-11 13:12:22 +01:00
Remi Cadene 5494faf096 fix unit tests (minimal) 2024-07-11 00:24:58 +02:00
Remi Cadene 01f8cede0b Add koch to CI 2024-07-10 23:05:36 +02:00
Remi Cadene 5d092b1be2 nit 2024-07-10 20:01:29 +02:00
Remi Cadene 7560372404 removed to radian representations 2024-07-10 19:56:51 +02:00
Remi Cadene 1bc13f39bf rename color to color_mode 2024-07-10 19:54:23 +02:00
Remi Cadene 95de4b7454 Skip tests if alexander koch robot is not available 2024-07-10 19:41:57 +02:00
Remi Cadene c2388c59be uncomment shutil.rmtree(tmp_imgs_dir) 2024-07-10 19:07:15 +02: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
Remi Cadene f7a7ed9aa9 Update opencv and dynamixel (find_port) 2024-07-10 18:47:59 +02:00
Remi Cadene 7411cf2a7a Improve opencv 2024-07-10 17:43:33 +02:00
Remi Cadene 2bebdf78a0 Nit + Remove benchmark opencv + Clean + Add docstring 2024-07-10 17:35:18 +02:00
Remi Cadene fb06417a83 remove comment read/write single value 2024-07-10 14:13:22 +02:00
Remi Cadene dd7bce7563 format 2024-07-10 14:11:53 +02:00
Remi Cadene 68a561570c fix unit test 2024-07-10 14:05:58 +02:00
Remi Cadene 52e760a88e Almost done 2024-07-10 00:07:40 +02:00
Remi Cadene 798373e7bf All tests passing except test_control_robot.py 2024-07-09 22:57:36 +02:00
Remi Cadene a0432f1608 Add unit tests (WIP) 2024-07-09 22:57:36 +02:00
Remi Cadene 3ba05d53b9 Remove async in dynamixel 2024-07-09 22:57:36 +02:00
Remi Cadene 72b2e342ae revert threads to thread, results to color_image in OpenCVCamera 2024-07-09 22:57:36 +02:00
Remi Cadene d83d34d9b3 Add log_control_info 2024-07-09 22:57:36 +02:00
Remi Cadene 3ff789c181 Add async_read and async_write (it doesnt work) 2024-07-09 22:57:36 +02:00
Remi Cadene 6e77a399a2 Add dynamixel-sdk as extra 2024-07-09 22:57:35 +02:00
Remi Cadene 8a7aa50e97 Style 2024-07-09 22:57:12 +02:00
Remi Cadene 47aac0dff7 rebase lock 2024-07-09 22:57:11 +02:00
Remi Cadene 858d49fc04 Add robot_devices and control_robot script 2024-07-09 22:56:46 +02: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
Alexander Soare 7bd5ab16d1
Fix generation of dataset test artifact (#306) 2024-07-05 11:02:26 +01:00
Simon Alibert 74362ac453
Add VQ-BeT copyrights (#299) 2024-07-04 13:02:31 +02:00
Simon Alibert 964f9e86d6
Cleanup config defaults (#300) 2024-07-04 11:53:29 +02:00
Nur Muhammad "Mahi" Shafiullah 7a5fc76b9f
Added new credits and citations (#301) 2024-07-03 20:57:47 +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
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