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:"创建时间"`
}