Spaces:
Sleeping
Sleeping
Commit
·
11c6b76
1
Parent(s):
b904b20
limit steps
Browse files
app.py
CHANGED
@@ -489,7 +489,7 @@ def run_training():
|
|
489 |
per_device_eval_batch_size=4,
|
490 |
gradient_accumulation_steps=8,
|
491 |
num_train_epochs=3,
|
492 |
-
max_steps=
|
493 |
learning_rate=3e-4,
|
494 |
weight_decay=0.01,
|
495 |
logging_steps=5, # Giảm số bước logging để theo dõi thường xuyên hơn
|
|
|
489 |
per_device_eval_batch_size=4,
|
490 |
gradient_accumulation_steps=8,
|
491 |
num_train_epochs=3,
|
492 |
+
max_steps=5, # Đặt max_steps tại đây
|
493 |
learning_rate=3e-4,
|
494 |
weight_decay=0.01,
|
495 |
logging_steps=5, # Giảm số bước logging để theo dõi thường xuyên hơn
|