lerobot/lerobot/common/policies
IrvingF7 f15dd56d0e
modified loss.item to loss.detach, so that no scalar will be created and when using torch.compile, it will not break graph
2025-02-14 08:43:09 -05:00
..
act Remove offline training, refactor `train.py` and logging/checkpointing (#670) 2025-02-11 10:36:06 +01:00
diffusion Remove offline training, refactor `train.py` and logging/checkpointing (#670) 2025-02-11 10:36:06 +01:00
pi0 modified loss.item to loss.detach, so that no scalar will be created and when using torch.compile, it will not break graph 2025-02-14 08:43:09 -05:00
tdmpc Remove offline training, refactor `train.py` and logging/checkpointing (#670) 2025-02-11 10:36:06 +01:00
vqbet Remove offline training, refactor `train.py` and logging/checkpointing (#670) 2025-02-11 10:36:06 +01:00
__init__.py Add Pi0 (#681) 2025-02-04 18:01:04 +01:00
factory.py Add Pi0 (#681) 2025-02-04 18:01:04 +01:00
normalize.py Add Pi0 (#681) 2025-02-04 18:01:04 +01:00
pretrained.py Remove offline training, refactor `train.py` and logging/checkpointing (#670) 2025-02-11 10:36:06 +01:00
utils.py Simplify configs (#550) 2025-01-31 13:57:37 +01:00