update limit feature steps

checkpoint.py

@@ -120,11 +120,11 @@ def run_training():
         max_steps=50,  # Đặt max_steps tại đây
         learning_rate=3e-4,
         weight_decay=0.01,
-        logging_steps=10,  # Giảm số bước logging để theo dõi thường xuyên hơn
+        logging_steps=5,  # Giảm số bước logging để theo dõi thường xuyên hơn
         eval_strategy="steps",  # Đánh giá sau mỗi vài bước
-        eval_steps=50,  # Đánh giá sau mỗi 50 bước
+        eval_steps=5,  # Đánh giá sau mỗi 50 bước
         save_strategy="steps",  # Lưu checkpoint sau mỗi vài bước
-        save_steps=50,  # Lưu checkpoint sau mỗi 50 bước
+        save_steps=5,  # Lưu checkpoint sau mỗi 50 bước
         save_total_limit=5,  # Giới hạn số lượng checkpoint lưu trữ
         fp16=True,
         report_to="none",
@@ -163,7 +163,7 @@ def run_training():
     return "Huấn luyện hoàn tất hoặc đã tiếp tục từ checkpoint."
 
 # Hàm Tự Động Hóa Việc Gọi Lặp Lại Hàm Huấn Luyện
-def continuous_training(total_steps=300, steps_per_call=50):
+def continuous_training(total_steps=300, steps_per_call=5):
     """
     Hàm tự động gọi lại `run_training` để hoàn thành quá trình huấn luyện.
     
@@ -187,6 +187,6 @@ def continuous_training(total_steps=300, steps_per_call=50):
         # Chờ một khoảng thời gian trước khi gọi lại (tùy thuộc vào yêu cầu của hệ thống)
         time.sleep(2)  # Thời gian chờ có thể điều chỉnh
 
-# Gọi hàm huấn luyện liên tục
-if __name__ == "__main__":
-    continuous_training(total_steps=300, steps_per_call=50)
+# # Gọi hàm huấn luyện liên tục
+# if __name__ == "__main__":
+#     continuous_training(total_steps=300, steps_per_call=10)