tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Fix validation for early stopping (#1358)

Browse Source
This commit is contained in:
Chirag Jain
2024-03-04 08:45:18 +05:30
committed by GitHub
parent 170d4d7092
commit b5b44925ec
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/utils/config/models/input/v0_4_1/__init__.py
View File

@@ -822,7 +822,7 @@ class AxolotlInputConfig(
@model_validator(mode="after") @model_validator(mode="after")
def check_early_stopping(self): def check_early_stopping(self):
if self.early_stopping_patience: if self.early_stopping_patience:
if not self.save_steps or self.eval_steps: if not self.save_steps or not self.eval_steps:
raise ValueError( raise ValueError(
"`early_stopping_patience` requires save_steps and eval_steps to be set. eval_steps should evenly divide save_steps." "`early_stopping_patience` requires save_steps and eval_steps to be set. eval_steps should evenly divide save_steps."
) )