Spaces:
Runtime error
Runtime error
Update train.py
Browse files
train.py
CHANGED
@@ -181,7 +181,7 @@ def train_model(model, tokenizer, dataset, push, isinst):
|
|
181 |
save_steps=WARMUP_STEPS * 5,
|
182 |
logging_steps=WARMUP_STEPS,
|
183 |
eval_strategy="no",
|
184 |
-
eval_steps=
|
185 |
save_total_limit=2,
|
186 |
)
|
187 |
|
|
|
181 |
save_steps=WARMUP_STEPS * 5,
|
182 |
logging_steps=WARMUP_STEPS,
|
183 |
eval_strategy="no",
|
184 |
+
# eval_steps=WARMUP_STEPS,
|
185 |
save_total_limit=2,
|
186 |
)
|
187 |
|