admin
a0b115d954
- 标注：AI 预标注直写 labels_json（去候选确认两阶段）；重叠去重（minIoU）；全量标注按钮 - 训练：脚本迁移入 server/training/（Go 化 prepare_yolo/analyze_rfdetr，保留 train_server.py）；tflite 产物自检并入训练流程（check_tflite） - 数据目录/权重不进 git；.gitignore 迁移至仓库根
26 lines
1.9 KiB
Go
26 lines
1.9 KiB
Go
package entity
|
|
|
|
import "github.com/gogf/gf/v2/os/gtime"
|
|
|
|
// ModelTraining 训练任务：runner 启动训练进程，轮询解析 epoch 日志更新进度/指标，
|
|
// 日志尾部截断存 log_tail；pid 用于取消与存活探测。
|
|
type ModelTraining struct {
|
|
Id int64 `json:"id" orm:"id" description:"自增主键"`
|
|
Name string `json:"name" orm:"name" description:"任务名"`
|
|
Status string `json:"status" orm:"status" description:"running|success|failed"`
|
|
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:"创建时间"`
|
|
}
|