Update train_cifar10.py

kentaroy47 · Jun 23, 2022 · ad0e19a · ad0e19a
1 parent 8e209cd
commit ad0e19a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_cifar10.py b/train_cifar10.py
@@ -32,7 +32,7 @@
 
 # parsers
 parser = argparse.ArgumentParser(description='PyTorch CIFAR10 Training')
-parser.add_argument('--lr', default=1e-3, type=float, help='learning rate') # resnets.. 1e-3, Vit..1e-4?
+parser.add_argument('--lr', default=1e-4, type=float, help='learning rate') # resnets.. 1e-3, Vit..1e-4
 parser.add_argument('--opt', default="adam")
 parser.add_argument('--resume', '-r', action='store_true', help='resume from checkpoint')
 parser.add_argument('--noaug', action='store_true', help='disable use randomaug')