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 ------------------------------------------------------------------