Deep CFR 무제한 학습 config 추가

This commit is contained in:
2026-05-07 03:52:59 +09:00
parent 148be6e9a0
commit 5fa4bd65ef
3 changed files with 97 additions and 1 deletions
@@ -18,7 +18,7 @@ class StrictModel(BaseModel):
class RunConfig(StrictModel):
experiment_name: str = "deep_cfr"
iterations: int = 1
iterations: int | None = 1
max_iterations: int | None = None
max_hours: float | None = None
seed: int = 1
@@ -534,6 +534,8 @@ class DeepCFRTrainer:
return max(self.iteration, int(self.config.run.max_iterations))
if self.config.run.max_hours is not None:
return 2**31 - 1
if self.config.run.iterations is None:
return 2**31 - 1
return self.iteration + self.config.run.iterations
def _time_limit_reached(self, run_started: float) -> bool: