From ad0e19a923539086ee83f42e7fa35c0d3a7a44f3 Mon Sep 17 00:00:00 2001 From: Kentaro Yoshioka Date: Thu, 23 Jun 2022 09:44:24 +0900 Subject: [PATCH] Update train_cifar10.py --- train_cifar10.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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')