Update train.py

Dataset
Zi-YuanYang · Mar 19, 2024 · 461f9cc · 461f9cc
1 parent 2bf70da
commit 461f9cc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train.py b/train.py
@@ -383,7 +383,7 @@ def fit(epoch, model, data_loader, phase='training'):
         epoch_loss, epoch_accuracy = fit(epoch, net, data_loader_train, phase='training')
         # print('finish')
 
-        val_epoch_loss, val_epoch_accuracy = fit(epoch, net, data_loader_test, phase='testing')
+        val_epoch_loss, val_epoch_accuracy = fit(epoch, net, data_loader_train, phase='testing')
 
         scheduler.step()
 
@@ -419,4 +419,4 @@ def fit(epoch, model, data_loader, phase='training'):
 
     print('------------\n')
     print('Best')
-    test(best_net)
+    test(best_net)