Feature remove code (#50)

Remove redundant code 'net.train()'
AlexHex7 · Aug 23, 2022 · 57f5230 · 57f5230
1 parent 9089828
commit 57f5230
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -90,6 +90,8 @@ to **Non-Local_pytorch_0.3.1/**.
 
 16. Verify that the code works well in **pytorch 1.12.0**.
 
+17. Remove redundant the code `net.train()` in training code files.
+
 ## Todo
 - Experiments on Charades dataset.
 - Experiments on COCO dataset.

diff --git a/demo_MNIST_AMP_train_with_single_gpu.py b/demo_MNIST_AMP_train_with_single_gpu.py
@@ -82,8 +82,6 @@
         total_acc += acc
         total_sample += img_batch.size(0)
 
-    net.train()
-
     mean_acc = total_acc.item() * 1.0 / total_sample
     mean_loss = sum(total_loss) / total_loss.__len__()
 

diff --git a/demo_MNIST_train.py b/demo_MNIST_train.py
@@ -67,8 +67,6 @@
         total_acc += acc
         total_sample += img_batch.size(0)
 
-    net.train()
-
     mean_acc = total_acc.item() * 1.0 / total_sample
     mean_loss = sum(total_loss) / total_loss.__len__()