Commit Graph

612 Commits

Author SHA1 Message Date
Seungjae Lee c0bed3e841
Merge pull request #5 from huggingface/main
Pull main branch
2024-05-27 13:45:19 -04: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
jayLEE0301 fd95f2e89b fix NT assignment error 2024-05-25 16:52:38 -04:00
jayLEE0301 5f6b66562f fix typo 2024-05-24 20:04:14 -04:00
jayLEE0301 547d3c3ea1 add comments, add primary_code_loss_weight 2024-05-24 18:06:12 -04:00
jayLEE0301 71ec76fe2a add explanations of each part of vq-bet 2024-05-24 15:58:46 -04:00
jayLEE0301 d71db341bc replace hardcoded part with gpt_num_obs_mode, delete unused eos token 2024-05-24 15:46:29 -04:00
jayLEE0301 b3e0ec1afd add bin pred temperature 2024-05-24 15:44:18 -04:00
jayLEE0301 8ee1e53fee remove redundant comment, change scheduler name 2024-05-24 14:29:00 -04:00
jayLEE0301 0b663243e3 remove _ in the names of models 2024-05-24 14:20:58 -04:00
Alexander Soare 6d39b73399
Adds a tutorial section on how to use arbitrary configuration files (#206) 2024-05-24 12:39:11 +01:00
jayLEE0301 340f7cfd6e add comment, change param names, delete kwargs 2024-05-23 19:08:37 -04:00
jayLEE0301 e301caf182 remove unused parameter explanations, fix some names of parameters 2024-05-23 15:43:04 -04:00
jayLEE0301 d209c0f73c delete redundant kwargs, change parameter names 2024-05-23 10:25:35 -04: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
Seungjae Lee 2f4f137586
Merge pull request #4 from huggingface/main
merge changes from HF repo
2024-05-22 18:34:09 -04:00
Seungjae Lee f127837d48
Merge pull request #3 from jayLEE0301/debugging-vqbet
Debugging vqbet
2024-05-22 18:25:16 -04:00
jayLEE0301 e57efff4c1 minor 2024-05-22 18:23:21 -04:00
jayLEE0301 d8f8fa5918 add warnings, merge to main branch 2024-05-22 18:15:45 -04:00
jayLEE0301 1d689512af clean code 2024-05-22 17:50:37 -04: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
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
jayLEE0301 311f79a874 split img emb / state 2024-05-13 20:33:58 -04:00
jayLEE0301 02d55c0b9a add temporary eval part, optimizer 2024-05-13 20:31:06 -04:00
Simon Alibert 89c6be84ca
Limit datasets major update (#176)
Co-authored-by: Quentin Lhoest <42851186+lhoestq@users.noreply.github.com>
Co-authored-by: Quentin Lhoest <lhoest.q@gmail.com>
2024-05-12 08:15:07 +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
Simon Alibert 29a196c5dd
Fix #173 - Require gym-pusht to be installed for test_examples_3_and_2 (#174) 2024-05-12 08:08:59 +02:00
Remi ced3de4c94
Fix hanging in visualize_dataset.py when num_workers > 0 (#165) 2024-05-11 19:28:22 +03:00
Vincent Moens 7b47ab211b
Remove torchrl acknowledgement (#177) 2024-05-11 14:45:51 +03: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
Seungjae Lee 23fbc1959c
Merge pull request #2 from huggingface/main
Merge from main
2024-05-09 20:07:28 -04:00
jayLEE0301 44edc6f648 Consistency with other models's configs 2024-05-09 20:04:51 -04:00
jayLEE0301 db4e5e7ec1 Let vqvae inherit from nn.module 2024-05-09 20:01:48 -04:00
Alexander Soare b187942db4
Add context manager for seeding (#164) 2024-05-09 17:58:39 +01:00
Alexander Soare 473345fdf6
Fix stats override in ACT config (#161) 2024-05-09 15:16:47 +01:00
Alexander Soare e89521dfa0
Enable tests for TD-MPC (#160) 2024-05-09 13:42:12 +01:00
Seungjae Lee 00d2422710
Merge pull request #1 from huggingface/main
Merge from main
2024-05-08 16:47:45 -04:00