lerobot/lerobot
AdilZouitine 550a4cf223 Remove torch.no_grad decorator and optimize next action prediction in SAC policy
- Removed `@torch.no_grad` decorator from Unnormalize forward method

- Added TODO comment for optimizing next action prediction in SAC policy
- Minor formatting adjustment in NaN assertion for log standard deviation
Co-authored-by: Yoel Chornton <yoel.chornton@gmail.com>
2025-03-24 09:40:36 +00:00
..
common Remove torch.no_grad decorator and optimize next action prediction in SAC policy 2025-03-24 09:40:36 +00:00
configs [Port HIL-SERL] Balanced sampler function speed up and refactor to align with train.py (#715) 2025-03-24 09:40:36 +00:00
scripts [Port HIL-SERL] Balanced sampler function speed up and refactor to align with train.py (#715) 2025-03-24 09:40:36 +00:00
templates Add Pi0 (#681) 2025-02-04 18:01:04 +01:00
__init__.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2025-03-24 09:40:36 +00:00
__version__.py Add copyrights (#157) 2024-05-15 12:13:09 +02:00