Alexander Soare
|
e01558a432
|
Merge remote-tracking branch 'upstream/main' into train_tdmpc
|
2024-05-22 16:15:03 +01: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
|
0905cf5cf1
|
Merge remote-tracking branch 'origin/train_tdmpc' into train_tdmpc
|
2024-05-21 08:17:09 +01:00 |
|
Alexander Soare
|
0f85bf5397
|
Merge branch 'fix_online_data' into train_tdmpc
|
2024-05-21 08:16:38 +01:00 |
|
Alexander Soare
|
fa2a418067
|
Merge remote-tracking branch 'upstream/main' into fix_online_data
|
2024-05-21 08:14:21 +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 |
|
Alexander Soare
|
99243b0343
|
Merge remote-tracking branch 'upstream/main' into train_tdmpc
|
2024-05-20 18:26:59 +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
|
2169a8d146
|
backup wip
|
2024-05-17 16:52:59 +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 |
|
Alexander Soare
|
918868162e
|
backup
|
2024-05-16 09:51:01 +01:00 |
Simon Alibert
|
f52f4f2cd2
|
Add copyrights (#157)
|
2024-05-15 12:13:09 +02: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 |
|
Alexander Soare
|
fb202b5040
|
backup wip
|
2024-05-10 07:07:55 +01:00 |
|
Alexander Soare
|
77aa80e198
|
Merge remote-tracking branch 'upstream/main' into train_tdmpc
|
2024-05-09 17:58:55 +01: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
|
ad49a4f3f5
|
backup wip
|
2024-05-09 13:53:20 +01:00 |
|
Alexander Soare
|
cce0c53d95
|
Merge remote-tracking branch 'upstream/main' into train_tdmpc
|
2024-05-09 13:48:43 +01:00 |
|
Alexander Soare
|
e89521dfa0
|
Enable tests for TD-MPC (#160)
|
2024-05-09 13:42:12 +01:00 |
|
Alexander Soare
|
d767fdf958
|
ready for review
|
2024-05-09 12:22:43 +01:00 |
|
Alexander Soare
|
266da63279
|
Merge remote-tracking branch 'upstream/main' into train_tdmpc
|
2024-05-09 12:13:34 +01:00 |
|
Simon Alibert
|
7bb5b15f4c
|
Remove dependencies upper bounds constraints (#145)
|
2024-05-08 17:23:10 +00:00 |
|
Simon Alibert
|
df914aa76c
|
Update dev docker build (#148)
|
2024-05-08 17:21:58 +00: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 |
|
Alexander Soare
|
47de07658c
|
Override pretrained model config (#147)
|
2024-05-08 12:56:21 +01:00 |
|
Alexander Soare
|
d36ca387e8
|
backup wip
|
2024-05-08 11:31:54 +01:00 |
|
Alexander Soare
|
d7ffcc9127
|
Merge remote-tracking branch 'upstream/main' into train_tdmpc
|
2024-05-07 09:24:08 +01:00 |
|
Alexander Soare
|
26d9a070d8
|
Add policy and dataset tags to WandB logging (#144)
|
2024-05-07 09:09:21 +01:00 |
|
Alexander Soare
|
1e2cabd4e0
|
Merge branch 'cache_dataset' into train_tdmpc
|
2024-05-07 09:07:15 +01:00 |
|
Alexander Soare
|
9e8501ffa8
|
cherry pick dataset cache
|
2024-05-07 07:52:05 +01:00 |
|
Alexander Soare
|
7033926d04
|
add cache is not None
|
2024-05-06 17:54:36 +01:00 |
|
Alexander Soare
|
bca0f87b07
|
Merge remote-tracking branch 'upstream/main' into cache_dataset
|
2024-05-06 17:50:05 +01:00 |
|
Alexander Soare
|
84b3f6b06a
|
Have a caching option in the dataset
|
2024-05-06 17:37:14 +01:00 |