From e6fed94443177826cf7497a85e9cdcfc7c43ee21 Mon Sep 17 00:00:00 2001
From: cloudroam <cloudroam>
Date: 星期一, 21 四月 2025 16:49:49 +0800
Subject: [PATCH] fix

---
 train_repayment.py |    2 +-
 1 files changed, 1 insertions(+), 1 deletions(-)

diff --git a/train_repayment.py b/train_repayment.py
index a5ef722..9620d19 100644
--- a/train_repayment.py
+++ b/train_repayment.py
@@ -138,7 +138,7 @@
         evaluation_strategy="steps",
         eval_steps=RepaymentNERConfig.EVAL_STEPS,
         save_strategy="steps",
-        save_steps=RepaymentNERConfig.SAVE_STEPS,
+        save_steps=RepaymentNERConfig.EVAL_STEPS,
         save_total_limit=RepaymentNERConfig.SAVE_TOTAL_LIMIT,
         load_best_model_at_end=True,
         metric_for_best_model="overall_f1",

--
Gitblit v1.9.3