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

完成训练模块的转移

Browse Source
This commit is contained in:
sunyg
2025-04-17 11:03:05 +08:00
parent 4439687870
commit 74e8f0d415
188 changed files with 32931 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
apps/business/train/schemas/project_train.py
View File

@ -6,27 +6,29 @@
# @IDE : PyCharm
# @desc : pydantic 模型，用于数据库序列化操作
from datetime import datetime
from pydantic import BaseModel, Field, ConfigDict
from core.data_types import DatetimeStr
from typing import Optional
"""
项目训练版本信息表
"""
class ProjectTrain(BaseModel):
project_id: int = Field(..., title="None")
train_version: str = Field(..., title="None")
train_url: str = Field(..., title="None")
train_data: str = Field(..., title="None")
weights_id: int = Field(..., title="None")
weights_name: str = Field(..., title="None")
epochs: int = Field(..., title="None")
patience: int = Field(..., title="None")
best_pt: str = Field(..., title="None")
last_pt: str = Field(..., title="None")
user_id: int = Field(..., title="None")
class ProjectTrainIn(BaseModel):
project_id: Optional[int] = Field(..., description="项目id")
weights_id: Optional[str] = Field(None, description="权重文件")
epochs: Optional[int] = Field(50, description="训练轮数")
patience: Optional[int] = Field(20, description="早停的耐心值")
class ProjectTrainSimpleOut(ProjectTrain):
class ProjectTrainOut(BaseModel):
id: Optional[int] = Field(None, description="训练id")
train_version: Optional[str] = Field(None, description="训练版本号")
weights_name: Optional[str] = Field(None, description="权重名称")
epochs: Optional[int] = Field(None, description="训练轮数")
patience: Optional[int] = Field(None, description="早停的耐心值")
create_time: Optional[datetime] = Field(None, description="训练时间")
model_config = ConfigDict(from_attributes=True)
id: int = Field(..., title="编号")
create_datetime: DatetimeStr = Field(..., title="创建时间")
update_datetime: DatetimeStr = Field(..., title="更新时间")