Michel Aractingi d2c41b35db - 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-04-18 15:04:13 +02:00

23 KiB

Raw Blame History

View Raw