Deep CFR 무제한 학습 config 추가
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user