訓練進度測試

models/Model.go

@@ -17,8 +17,10 @@ type Model struct {
 	Progress     int       `json:"progress"`                 // (0-100)
 	Image        string    `json:"image"`                    // 封面圖片實際地址
 	Hash         string    `json:"hash"`                     // 模型哈希值
-	Tags         string    `json:"tags"`
+	Epochs       int       `json:"epochs"`                   // 訓練步數
+	Tags         TagList   `json:"tags"`
 	UserID       int       `json:"user_id"`
+	DatasetID    int       `json:"dataset_id"`
 	CreatedAt    time.Time `json:"created_at" gorm:"autoCreateTime"`
 	UpdatedAt    time.Time `json:"updated_at" gorm:"autoUpdateTime"`
 }