admin.suherdy/ai_github_trainer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix eval strategy error

Browse Source
This commit is contained in:
Suherdy Yacob 2025-08-22 20:05:43 +07:00
parent c7a84c520c
commit 8fc65ddae8
5 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
configs/training_config.yaml
View File

@ -26,7 +26,7 @@ training:
save_total_limit: 3
# Evaluation
evaluation_strategy: "steps"
eval_strategy: "steps"
eval_steps: 100
load_best_model_at_end: true
metric_for_best_model: "loss"

2
configs/training_config_qwen3.yaml
View File

@ -26,7 +26,7 @@ training:
save_total_limit: 3
# Evaluation
evaluation_strategy: "steps"
eval_strategy: "steps"
eval_steps: 100
load_best_model_at_end: true
metric_for_best_model: "loss"

BIN
logs/events.out.tfevents.1755867795.Suherdy-Laptop.20720.0 Normal file
View File

Binary file not shown.

4
src/config.py
View File

@ -33,7 +33,7 @@ class TrainingConfig:
logging_steps: int = 1
save_steps: int = 100
save_total_limit: int = 3
evaluation_strategy: str = "steps"
eval_strategy: str = "steps"
eval_steps: int = 100
load_best_model_at_end: bool = True
metric_for_best_model: str = "loss"
@ -187,7 +187,7 @@ class AppConfig:
'logging_steps': self.training.logging_steps,
'save_steps': self.training.save_steps,
'save_total_limit': self.training.save_total_limit,
'evaluation_strategy': self.training.evaluation_strategy,
'eval_strategy': self.training.eval_strategy,
'eval_steps': self.training.eval_steps,
'load_best_model_at_end': self.training.load_best_model_at_end,
'metric_for_best_model': self.training.metric_for_best_model,

2
src/trainer.py
View File

@ -189,7 +189,7 @@ class ModelTrainer:
logging_steps=self.config.training.logging_steps,
save_steps=self.config.training.save_steps,
save_total_limit=self.config.training.save_total_limit,
evaluation_strategy=self.config.training.evaluation_strategy,
eval_strategy=self.config.training.eval_strategy,
eval_steps=self.config.training.eval_steps,
load_best_model_at_end=self.config.training.load_best_model_at_end,
metric_for_best_model=self.config.training.metric_for_best_model,