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
This commit is contained in:
AdilZouitine
2025-03-04 13:22:35 +00:00
committed by Michel Aractingi
parent ae51c19b3c
commit bb69cb3c8c
3 changed files with 11 additions and 5 deletions

View File

@@ -64,6 +64,8 @@ class SACConfig:
}
)
camera_number: int = 1
storage_device: str = "cpu"
# Add type annotations for these fields:
vision_encoder_name: str | None = field(default="helper2424/resnet10")
freeze_vision_encoder: bool = True