28 lines
2.1 KiB
Go
28 lines
2.1 KiB
Go
package entity
|
||
|
||
import "github.com/gogf/gf/v2/os/gtime"
|
||
|
||
// ModelTraining 训练任务：runner 启动训练进程，轮询解析 epoch 日志更新进度/指标，
|
||
// 日志尾部截断存 log_tail；pid 用于取消与存活探测。
|
||
// 状态机 queued → running → success/failed（queued=GPU 忙排队，轮询晋级启动）。
|
||
type ModelTraining struct {
|
||
Id int64 `json:"id" orm:"id" description:"自增主键"`
|
||
Name string `json:"name" orm:"name" description:"任务名"`
|
||
Status string `json:"status" orm:"status" description:"queued|running|success|failed"`
|
||
Variant string `json:"variant" orm:"variant" description:"档位 s(高识别)|n(高性能)"`
|
||
DatasetId int64 `json:"datasetId" orm:"dataset_id" description:"来源数据集"`
|
||
Imgsz int `json:"imgsz" orm:"imgsz" description:"训练分辨率"`
|
||
Epochs int `json:"epochs" orm:"epochs" description:"目标轮数"`
|
||
Batch int `json:"batch" orm:"batch" description:"batch size"`
|
||
Device string `json:"device" orm:"device" description:"训练设备"`
|
||
CurrentEpoch int `json:"currentEpoch" orm:"current_epoch" description:"当前轮数"`
|
||
TotalEpochs int `json:"totalEpochs" orm:"total_epochs" description:"实际总轮数(完成时写)"`
|
||
Metrics string `json:"metrics" orm:"metrics" description:"JSON {p,r,map50}"`
|
||
LogTail string `json:"logTail" orm:"log_tail" description:"日志尾部(截断)"`
|
||
Pid int `json:"pid" orm:"pid" description:"训练进程 pid"`
|
||
Error string `json:"error" orm:"error" description:"失败原因"`
|
||
StartedAt *gtime.Time `json:"startedAt" orm:"started_at" description:"开始时间"`
|
||
FinishedAt *gtime.Time `json:"finishedAt" orm:"finished_at" description:"结束时间"`
|
||
CreatedAt *gtime.Time `json:"createdAt" orm:"created_at" description:"创建时间"`
|
||
}
|