yolo中断训练功能添加

2022-11-15 17:59:06 +08:00
parent b70ba8a431
commit cdb081654a
4 changed files with 46 additions and 8 deletions
--- a/app/schemas/TrainResult.py
+++ b/app/schemas/TrainResult.py
@ -32,6 +32,7 @@ class Report(BaseModel):
    train_mod_savepath: str = Field(..., description="模型保存路径")
    start_time: datetime.date = Field(datetime.datetime.now(), description="开始时间")
    end_time: datetime.date = Field(datetime.datetime.now(), description="结束时间")
+    alg_code: str = Field(..., description="模型编码")


 class ReportDict(BaseModel):