From 9818a1d488cb44946a07d5f5d6ed68b35d26f91b Mon Sep 17 00:00:00 2001
From: JIAKUNHAO <jiakunhao@star-rising.com>
Date: Wed, 23 Nov 2022 14:31:00 +0800
Subject: [PATCH] =?UTF-8?q?=E8=BF=94=E5=9B=9E=E8=AE=AD=E7=BB=83=E7=BB=88?=
 =?UTF-8?q?=E6=AD=A2=E4=BF=A1=E5=8F=B7=20isfinish=3D2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/yolov5/train_server.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/app/yolov5/train_server.py b/app/yolov5/train_server.py
index ab1872d..3a18779 100644
--- a/app/yolov5/train_server.py
+++ b/app/yolov5/train_server.py
@@ -326,6 +326,8 @@ def train(hyp, opt, device, data_list,id,callbacks):  # hyp is path/to/hyp.yaml
         report.progress = (i + 1)
         report.end_time = datetime.now()
         report.precision[0].value.append(reportAccu)
+        if (i+1) == num_epochs:
+            report.isfinish = 2
         return report.dict()
     ###################结束#######################
     for epoch in range(start_epoch, epochs):  # epoch ------------------------------------------------------------------