Spaces:
Build error
Build error
Update trainer configuration in train.py to align evaluation strategy with save strategy. Set eval_steps to match save_steps for consistent evaluation frequency.
Browse files
train.py
CHANGED
@@ -207,8 +207,10 @@ def create_trainer(
|
|
207 |
fp16=not is_bfloat16_supported(),
|
208 |
bf16=is_bfloat16_supported(),
|
209 |
logging_steps=1,
|
210 |
-
|
211 |
save_steps=30,
|
|
|
|
|
212 |
save_total_limit=2,
|
213 |
optim="adamw_8bit",
|
214 |
weight_decay=0.01,
|
|
|
207 |
fp16=not is_bfloat16_supported(),
|
208 |
bf16=is_bfloat16_supported(),
|
209 |
logging_steps=1,
|
210 |
+
save_strategy="steps",
|
211 |
save_steps=30,
|
212 |
+
eval_strategy="steps", # Match save_strategy
|
213 |
+
eval_steps=30, # Match save_steps
|
214 |
save_total_limit=2,
|
215 |
optim="adamw_8bit",
|
216 |
weight_decay=0.01,
|