更新参数

Author: Kedreamix

CIFAR10_code/eval.py

@@ -32,7 +32,8 @@ def eval_top5(outputs, label):
     parser.add_argument('--cuda', action='store_true', default=False, help =' use GPU?')
     parser.add_argument('--batch-size', default=64, type=int, help = "Batch Size for Test")
     parser.add_argument('--num-workers', default=2, type=int, help = 'num-workers')
-    parser.add_argument('--net', type = str, default='MobileNetv1', help='net type')
+    parser.add_argument('--net', type = str, choices=['LeNet5', 'AlexNet', 'VGG16','VGG19','ResNet18','ResNet34',   
+                                                       'DenseNet','MobileNetv1','MobileNetv2'], default='MobileNetv1', help='net type')
     args = parser.parse_args()
     testloader = get_test_dataloader()
 

CIFAR10_code/train.py

@@ -135,7 +135,8 @@ def train(epoch):
                                     'Lr'   : lr})
                 pbar.update(1)
         # train_loss = train_loss.item() / len(trainloader)
-        # train_acc = train_acc.item() * 100 / len(trainloader)    
+        # train_acc = train_acc.item() * 100 / len(trainloader)
+        scheduler.step(train_loss)
         print('Finish Train')
     def test(epoch):
         global best_acc
@@ -200,6 +201,6 @@ def test(epoch):
     for epoch in range(start_epoch, epochs):
         train(epoch)
         test(epoch)
-        scheduler.step()
+        
     torch.cuda.empty_cache()