完成线程启动训练并以websocket的方式进行发送shell

2025-03-10 17:42:56 +08:00
parent b4b1085403
commit 7d736c4ac4
9 changed files with 97 additions and 36 deletions
--- a/app/model/bussiness_model.py
+++ b/app/model/bussiness_model.py
@ -81,6 +81,10 @@ class ProjectTrain(DbCommon):
    __tablename__ = "project_train"
    project_id: Mapped[int] = mapped_column(Integer, nullable=False)
    train_version: Mapped[str] = mapped_column(String(32), nullable=False)
+    weights_id: Mapped[int] = mapped_column(Integer)
+    weights_name: Mapped[str] = mapped_column(String(32))
+    epochs: Mapped[int] = mapped_column(Integer)
+    patience: Mapped[int] = mapped_column(Integer)
    best_pt: Mapped[str] = mapped_column(String(255), nullable=False)
    last_pt: Mapped[str] = mapped_column(String(255), nullable=False)