fix function

config.py

@@ -1,12 +1,12 @@
 # config.py
 
 # --- 训练参数 ---
-LEARNING_RATE = 1e-4
-BATCH_SIZE = 4
-NUM_EPOCHS = 20
+LEARNING_RATE = 5e-5  # 降低学习率，提高训练稳定性
+BATCH_SIZE = 8  # 增加批次大小，提高训练效率
+NUM_EPOCHS = 50  # 增加训练轮数
 PATCH_SIZE = 256
-# (新增) 训练时尺度抖动范围
-SCALE_JITTER_RANGE = (0.7, 1.5) 
+# (优化) 训练时尺度抖动范围 - 缩小范围提高稳定性
+SCALE_JITTER_RANGE = (0.8, 1.2) 
 
 # --- 匹配与评估参数 ---
 KEYPOINT_THRESHOLD = 0.5