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
..
datasets	Reward classifier and training (#528)	2025-04-18 15:02:13 +02:00
envs	Stable version of rlpd + drq	2025-04-18 15:04:10 +02:00
optim	chore(doc): add license header to all files (#818)	2025-03-05 17:56:51 +01:00
policies	- Refactor observation encoder in `modeling_sac.py`	2025-04-18 15:04:13 +02:00
robot_devices	Extend reward classifier for multiple camera views (#626)	2025-04-18 15:03:50 +02:00
utils	[pre-commit.ci] pre-commit autoupdate (#871)	2025-04-08 06:58:46 +02:00
constants.py	chore(doc): add license header to all files (#818)	2025-03-05 17:56:51 +01:00
logger.py	Added support for checkpointing the policy. We can save and load the policy state dict, optimizers state, optimization step and interaction step	2025-04-18 15:04:13 +02:00