Added number of steps after success as parameter in config

This commit is contained in:
Michel Aractingi
2025-05-09 18:09:10 +02:00
committed by AdilZouitine
parent db86586530
commit aa793cbd4a
2 changed files with 7 additions and 6 deletions

View File

@@ -225,6 +225,9 @@ class HILSerlRobotEnvConfig(EnvConfig):
push_to_hub: bool = True
pretrained_policy_name_or_path: Optional[str] = None
reward_classifier_pretrained_path: Optional[str] = None
number_of_steps_after_success: int = (
0 # For the reward classifier, to record more positive examples after a success
)
def gym_kwargs(self) -> dict:
return {}