sun15684712291/aicheckv2
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

细化部分功能

Browse Source
This commit is contained in:
sunyg
2025-03-13 17:35:03 +08:00
parent 76f4a5ecd9
commit 6daea23f0a
7 changed files with 36 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
app/model/bussiness_model.py
View File

@ -82,6 +82,7 @@ class ProjectTrain(DbCommon):
project_id: Mapped[int] = mapped_column(Integer, nullable=False)
train_version: Mapped[str] = mapped_column(String(32), nullable=False)
train_url: Mapped[str] = mapped_column(String(255), nullable=False)
train_data: Mapped[str] = mapped_column(String(255), nullable=False)
weights_id: Mapped[int] = mapped_column(Integer)
weights_name: Mapped[str] = mapped_column(String(32))
epochs: Mapped[int] = mapped_column(Integer)

2
app/model/schemas/project_train_schemas.py
View File

@ -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="早停的耐心值")