This commit is contained in:
Botao Chen 2025-01-02 20:17:04 -08:00
parent e3f187fb83
commit f2a0468b2b
2 changed files with 3 additions and 1 deletions

View file

@ -58,6 +58,7 @@ class TrainingConfig(BaseModel):
n_epochs: int
max_steps_per_epoch: int
gradient_accumulation_steps: int
max_validation_steps: int
data_config: DataConfig
optimizer_config: OptimizerConfig
efficiency_config: Optional[EfficiencyConfig] = None