1

server/training/train_server.py

@@ -55,11 +55,11 @@ def on_fit_epoch_end(trainer):
     }, ensure_ascii=False) + "\n")
 
 
-def register_callback():
-    # ultralytics ≥8.4 中 callbacks 字典改由 get_default_callbacks() 取得
-    from ultralytics.utils.callbacks import get_default_callbacks
-    callbacks = get_default_callbacks()
-    callbacks["on_fit_epoch_end"].append(on_fit_epoch_end)
+def register_callback(model):
+    # ultralytics ≥8.4 中 trainer 使用 model 实例的 callbacks 字典（model.train 前以
+    # _callbacks=self.callbacks 传入），模块级 get_default_callbacks() 的 append 不生效；
+    # add_callback 追加到 model 字典即随训练触发，进度 jsonl 才有输出
+    model.add_callback("on_fit_epoch_end", on_fit_epoch_end)
 
 
 def write_result(result_file, payload):
@@ -245,8 +245,8 @@ def main():
 
     try:
         from ultralytics import YOLO
-        register_callback()
         model = YOLO("yolov8n.pt")
+        register_callback(model)
         model.train(
             data=data, imgsz=imgsz, epochs=epochs,
             patience=PATIENCE, batch=batch, device=device, workers=4,