AdilZouitine d8a1758122 Add storage device configuration for SAC policy and replay buffer ...

- Introduce `storage_device` parameter in SAC configuration and training settings
- Update learner server to use configurable storage device for replay buffer
- Reduce online buffer capacity in ManiSkill configuration
- Modify replay buffer initialization to support custom storage device

2025-03-04 13:22:35 +00:00

..

configuration_sac.py

Add storage device configuration for SAC policy and replay buffer

2025-03-04 13:22:35 +00:00

modeling_sac.py

Added caching function in the learner_server and modeling sac in order to limit the number of forward passes through the pretrained encoder when its frozen.

2025-02-21 10:13:43 +00:00