完成推理模块的主体功能

2025-03-04 17:04:37 +08:00
parent 4262d3e908
commit fa6c344e84
9 changed files with 325 additions and 21 deletions
--- a/app/service/project_train_service.py
+++ b/app/service/project_train_service.py
@ -212,7 +212,7 @@ def run_commend(data: str, project: str,
            train = ProjectTrain()
            train.project_id = project_id
            train.train_version = name
-            bast_pt_path = os.file_path(project, name, 'weights', 'bast.pt')
+            bast_pt_path = os.file_path(project, name, 'weights', 'best.pt')
            last_pt_path = os.file_path(project, name, 'weights', 'last.pt')
            train.best_pt = bast_pt_path
            train.last_pt = last_pt_path