1

server/common/training_runner.go

@@ -88,6 +88,10 @@ type TrainingConfig struct {
 	DatasetDir  string
 	Python      string
 	TimeoutMins int
+	Imgsz       int
+	Epochs      int
+	Batch       int
+	Device      string
 }
 
 // TrainingConfigOf 读取训练通道配置（未配置返回 ok=false）
@@ -97,6 +101,10 @@ 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(),
+		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(),
 	}
 	return cfg, cfg.Workdir != "" && cfg.Python != ""
 }