This commit is contained in:
2026-08-26 18:37:24 +08:00
parent 28220899fb
commit c323ca9031
10 changed files with 38 additions and 82 deletions
+3 -7
View File
@@ -207,15 +207,11 @@ type AdminTrainingListRes struct {
List []*AdminTrainingItem `json:"list"`
}
// AdminTrainingStartReq 发起训练(并发度 1:已有 running 任务时拒绝)
// AdminTrainingStartReq 发起训练(并发度 1:已有 running 任务时拒绝;训练参数走 config.yml training 节点,name 空则自动生成
type AdminTrainingStartReq struct {
g.Meta `path:"/trainings" method:"post" summary:"发起训练" tags:"管理端"`
DatasetId int64 `json:"datasetId" v:"required|min:1" dc:"数据集 id"`
Name string `json:"name" v:"required|length:1,50" dc:"任务名称"`
Imgsz int `json:"imgsz" v:"integer|min:64|max:2048" dc:"输入尺寸,默认 704"`
Epochs int `json:"epochs" v:"integer|min:1|max:1000" dc:"训练轮数,默认 150"`
Batch int `json:"batch" v:"integer|min:1|max:128" dc:"批大小,默认 16"`
Device string `json:"device" v:"length:0,16" dc:"设备,默认 0"`
DatasetId int64 `json:"datasetId" v:"required|min:1" dc:"数据集 id"`
Name string `json:"name" v:"length:0,50" dc:"任务名称,空自动生成"`
}
type AdminTrainingStartRes struct {