Merge branch 'master' of https://github.com/dragen1860/Deep-Learning-with-PyTorch-Tutorials

Author: dragen1860

README.md

@@ -1,15 +1,18 @@
+
+<!--  
 <p align="center">
   <a href="https://study.163.com/course/introduction/1208894818.htm?share=2&shareId=480000001847407
 ">
     <img src="res/ad_banner.png">
   </a>
 </p> 
-
+ -->
 **课程链接:** https://study.163.com/provider/480000001847407/index.htm?share=2&shareId=480000001847407
 
 <p align="center">
   <img width="300"  src="res/cover.png">
 </p> 
+
 
 **课程大纲:**
 ![课程介绍](res/outline.png)

lesson63-迁移学习-自定义数据集实战/train_scratch.py

@@ -29,6 +29,8 @@
 viz = visdom.Visdom()
 
 def evalute(model, loader):
+    model.eval()
+    
     correct = 0
     total = len(loader.dataset)
 
@@ -58,7 +60,8 @@ def main():
 
             # x: [b, 3, 224, 224], y: [b]
             x, y = x.to(device), y.to(device)
-
+            
+            model.train()
             logits = model(x)
             loss = criteon(logits, y)
 
@@ -94,4 +97,4 @@ def main():
 
 
 if __name__ == '__main__':
-    main()
+    main()

lesson63-迁移学习-自定义数据集实战/train_transfer.py

@@ -30,6 +30,8 @@
 viz = visdom.Visdom()
 
 def evalute(model, loader):
+    model.eval()
+    
     correct = 0
     total = len(loader.dataset)
 
@@ -68,6 +70,7 @@ def main():
             # x: [b, 3, 224, 224], y: [b]
             x, y = x.to(device), y.to(device)
 
+            model.train()
             logits = model(x)
             loss = criteon(logits, y)
 
@@ -103,4 +106,4 @@ def main():
 
 
 if __name__ == '__main__':
-    main()
+    main()