细化部分功能

2025-03-13 17:35:03 +08:00
parent 76f4a5ecd9
commit 6daea23f0a
7 changed files with 36 additions and 34 deletions
--- a/app/model/schemas/project_train_schemas.py
+++ b/app/model/schemas/project_train_schemas.py
@ -5,7 +5,7 @@ from typing import Optional

 class ProjectTrainIn(BaseModel):
    project_id: Optional[int] = Field(..., description="项目id")
-    weights_id: Optional[int] = Field(None, description="权重文件")
+    weights_id: Optional[str] = Field(None, description="权重文件")
    epochs: Optional[int] = Field(50, description="训练轮数")
    patience: Optional[int] = Field(20, description="早停的耐心值")