update curve

layumi · May 11, 2023 · 879baf5 · 879baf5
1 parent 99dfb15
commit 879baf5
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/train.py b/train.py
@@ -397,8 +397,7 @@ def train_model(model, criterion, optimizer, scheduler, num_epochs=25):
                     save_network(model.module, epoch+1)
                 else:
                     save_network(model, epoch+1)
-
-                draw_curve(epoch)
+            draw_curve(epoch)
             if phase == 'train':
                scheduler.step()
         time_elapsed = time.time() - since