Update train.py

Author: Kedreamix

CIFAR10_code/train.py

@@ -86,8 +86,7 @@
 
     early_stopping = EarlyStopping(patience = args.patience, verbose=True)
     criterion = nn.CrossEntropyLoss()
-    optimizer = optim.SGD(net.parameters(), lr=args.lr,
-                        momentum=0.9, weight_decay=5e-4)
+    optimizer = optim.AdamW(net.parameters(), lr=args.lr, momentum=0.9, weight_decay=0)
     scheduler = torch.optim.lr_scheduler.ReduceLROnPlateau(optimizer, 'min', factor=0.94,verbose=True,patience = 1,min_lr = 0.000001) # 动态更新学习率
 
     epochs = args.epochs