AdilZouitine e002c5ec56 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-04-18 15:06:52 +02:00

..

act

[pre-commit.ci] auto fixes from pre-commit.com hooks

2025-04-18 15:06:51 +02:00

diffusion

[pre-commit.ci] auto fixes from pre-commit.com hooks

2025-04-18 15:06:51 +02:00

hilserl

[pre-commit.ci] auto fixes from pre-commit.com hooks

2025-04-18 15:06:51 +02:00

pi0

feat(visualization): replace cv2 GUI with Rerun (and solves ffmpeg versioning issues) (#903)

2025-04-09 17:33:01 +02:00

pi0fast

Support for PI0+FAST (#921)

2025-04-04 11:51:11 +02:00

sac

Remove torch.no_grad decorator and optimize next action prediction in SAC policy

2025-04-18 15:06:52 +02:00

tdmpc

[pre-commit.ci] auto fixes from pre-commit.com hooks

2025-04-18 15:06:51 +02:00

vqbet

[pre-commit.ci] auto fixes from pre-commit.com hooks

2025-04-18 15:06:51 +02:00

__init__.py

chore(doc): add license header to all files (#818)

2025-03-05 17:56:51 +01:00

factory.py

Change SAC policy implementation with configuration and modeling classes

2025-04-18 15:03:51 +02:00

normalize.py

Remove torch.no_grad decorator and optimize next action prediction in SAC policy

2025-04-18 15:06:52 +02:00

pretrained.py

refactor(config): Move device & amp args to PreTrainedConfig (#812)

2025-03-06 17:59:28 +01:00

utils.py

Simplify configs (#550)

2025-01-31 13:57:37 +01:00