Add grasp critic

- Implemented grasp critic to evaluate gripper actions
- Added corresponding config parameters for tuning
This commit is contained in:
s1lent4gnt
2025-03-31 17:35:59 +02:00
committed by Adil Zouitine
parent 0f706ce543
commit 384eb2cd07
2 changed files with 132 additions and 0 deletions

View File

@@ -42,6 +42,14 @@ class CriticNetworkConfig:
final_activation: str | None = None
@dataclass
class GraspCriticNetworkConfig:
hidden_dims: list[int] = field(default_factory=lambda: [256, 256])
activate_final: bool = True
final_activation: str | None = None
output_dim: int = 3
@dataclass
class ActorNetworkConfig:
hidden_dims: list[int] = field(default_factory=lambda: [256, 256])