1

server/common/training_runner.go

@@ -64,7 +64,7 @@ type TrainingJob struct {
 	TaskId      int64
 	DatasetName string // yolo 数据集目录名（训练机 datasetDir/yolo/<name>）
 	Python      string // 训练机 venv python 路径
-	Workdir     string // 训练机工作目录（train_server.py / yolov8n.pt 所在）
+	Workdir     string // 训练机工作目录（train_server.py / yolov8s.pt 所在）
 	DatasetDir  string // 训练机数据集根目录（相对 workdir）
 }
 
@@ -87,6 +87,7 @@ type TrainingConfig struct {
 	DatasetDir  string
 	Python      string
 	TimeoutMins int
+	Model       string // 训练基座权重（训练机 workdir 下，如 yolov8s.pt）
 	Imgsz       int
 	Epochs      int
 	Batch       int
@@ -100,7 +101,8 @@ func TrainingConfigOf(ctx context.Context) (TrainingConfig, bool) {
 		DatasetDir:  g.Cfg().MustGet(ctx, "training.datasetDir", "datasets").String(),
 		Python:      g.Cfg().MustGet(ctx, "training.venvPython").String(),
 		TimeoutMins: g.Cfg().MustGet(ctx, "training.timeoutMinutes", 600).Int(),
-		Imgsz:       g.Cfg().MustGet(ctx, "training.imgsz", 704).Int(),
+		Model:       g.Cfg().MustGet(ctx, "training.model", "yolov8s.pt").String(),
+		Imgsz:       g.Cfg().MustGet(ctx, "training.imgsz", 1280).Int(),
 		Epochs:      g.Cfg().MustGet(ctx, "training.epochs", 150).Int(),
 		Batch:       g.Cfg().MustGet(ctx, "training.batch", 16).Int(),
 		Device:      g.Cfg().MustGet(ctx, "training.device", "0").String(),