diff --git a/train_cifar10.py b/train_cifar10.py index 4a8c00f..53cea32 100644 --- 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')