Commit 9b134f11 authored by zhiyang.zhou's avatar zhiyang.zhou

minor change

parent 998940bc
Pipeline #197 failed with stages
......@@ -18,7 +18,7 @@ parser.add_argument('--model_name', default='resnet-50',
parser.add_argument('--gpuid', default='0', type=str, help='which gpu to use')
parser.add_argument('--no_cuda', action='store_true', default=False, help='disables CUDA training')
parser.add_argument('-b', '--batch_size', default=32, type=int, help='mini-batch size (default: 32)')
parser.add_argument('--lr', type=float, default=0.05, metavar='LR', help='learning rate')
parser.add_argument('--lr', type=float, default=0.001, metavar='LR', help='learning rate')
parser.add_argument('--momentum', type=float, default=0.9, metavar='M', help='SGD momentum')
parser.add_argument('--weight_decay', '--wd', default=0, type=float, metavar='W')
parser.add_argument('--epochs', type=int, default=20, metavar='N', help='number of epochs to train')
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment