Michel Aractingi 506821c7df - Refactor observation encoder in modeling_sac.py ...

- added `torch.compile` to the actor and learner servers.
- organized imports in `train_sac.py`
- optimized the parameters push by not sending the frozen pre-trained encoder.

Co-authored-by: Adil Zouitine <adilzouitinegm@gmail.com>

2025-02-03 15:07:58 +00:00

..

act

fix(act): n_vae_encoder_layers config parameter wasn't being used (#400)

2024-09-02 18:29:27 +01:00

diffusion

feat: enable to use multiple rgb encoders per camera in diffusion policy (#484)

2024-10-30 11:00:05 +01:00

hilserl

Extend reward classifier for multiple camera views (#626)

2025-01-13 13:57:49 +01:00

sac

- Refactor observation encoder in modeling_sac.py

2025-02-03 15:07:58 +00:00

tdmpc

Improve discoverability on the hub (#325)

2024-08-19 15:16:46 +02:00

vqbet

Bug: Fix VQ-Bet not working when n_action_pred_token=1 (#420)

2024-09-09 09:41:13 +01:00

factory.py

Added support for checkpointing the policy. We can save and load the policy state dict, optimizers state, optimization step and interaction step

2025-02-03 15:07:58 +00:00

normalize.py

Make sure policies don't mutate the batch (#323)

2024-07-22 20:38:33 +01:00

policy_protocol.py

Improves Type Annotations (#252)

2024-06-10 19:09:48 +01:00

utils.py

Make policies compatible with other/multiple image keys (#149)

2024-05-16 13:51:53 +01:00