更新训练脚本，调整批量大小和工作线程数以提高训练效率

Mq-b · Mq-b · commit d06110218699 · 2026-04-09T21:40:25.000+08:00
diff --git a/py/train.py b/py/train.py
@@ -46,9 +46,9 @@
 RUN_NAME = "WellColumnClassification"
 IMG_SIZE = 224
 EPOCHS = 200
-BATCH_SIZE = 16
+BATCH_SIZE = 64
 DEVICE = 0
-WORKERS = 0
+WORKERS = 4
 
 
 def iter_class_dirs(root: Path) -> list[Path]:
