1

server/training/train_server.py

@@ -6,9 +6,10 @@
     python train_server.py --task-json tasks/<taskId>.json
 
 任务参数（Go 侧写入，字段相对训练机 workdir）：
-    workdir       训练机工作目录（脚本 / yolov8n.pt / venv 所在），启动即 chdir
+    workdir       训练机工作目录（脚本 / yolov8s.pt / venv 所在），启动即 chdir
     yolo          训练集目录（含 dataset.yaml），相对 workdir
-    imgsz         训练/导出分辨率（默认 704，与端侧推理对齐）
+    model         训练基座权重文件名（workdir 下，默认 yolov8s.pt）
+    imgsz         训练/导出分辨率（默认 1280，与端侧推理对齐）
     epochs        训练轮数
     batch         批大小
     device        GPU 编号或 cpu
@@ -236,7 +237,7 @@ def main():
     except Exception:
         pass  # 日志不可写（如 workdir 缺失）不阻断错误路径，traceback 仍走 stderr
 
-    imgsz = int(task.get("imgsz") or 704)
+    imgsz = int(task.get("imgsz") or 1280)
     epochs = int(task.get("epochs") or 150)
     batch = int(task.get("batch") or 16)
     device = task.get("device") or "0"
@@ -245,7 +246,7 @@ def main():
 
     try:
         from ultralytics import YOLO
-        model = YOLO("yolov8n.pt")
+        model = YOLO(task.get("model") or "yolov8s.pt")
         register_callback(model)
         model.train(
             data=data, imgsz=imgsz, epochs=epochs,