hoduyquocbao commited on
Commit
11c6b76
·
1 Parent(s): b904b20

limit steps

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=50, # Đặ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
 
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