forked from tangger/lerobot
Add grasp critic
- Implemented grasp critic to evaluate gripper actions - Added corresponding config parameters for tuning
This commit is contained in:
committed by
Adil Zouitine
parent
0f706ce543
commit
384eb2cd07
@@ -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])
|
||||
|
||||
Reference in New Issue
Block a user