|
|
|
@ -147,7 +147,7 @@ parser.add_argument('--warmup-lr', type=float, default=0.0001, metavar='LR',
|
|
|
|
|
parser.add_argument('--min-lr', type=float, default=1e-5, metavar='LR',
|
|
|
|
|
help='lower lr bound for cyclic schedulers that hit 0 (1e-5)')
|
|
|
|
|
parser.add_argument('--epochs', type=int, default=200, metavar='N',
|
|
|
|
|
help='number of epochs to train (default: 2)')
|
|
|
|
|
help='number of epochs to train (default: 200)')
|
|
|
|
|
parser.add_argument('--epoch-repeats', type=float, default=0., metavar='N',
|
|
|
|
|
help='epoch repeat multiplier (number of times to repeat dataset epoch per train epoch).')
|
|
|
|
|
parser.add_argument('--start-epoch', default=None, type=int, metavar='N',
|
|
|
|
@ -249,7 +249,7 @@ parser.add_argument('--recovery-interval', type=int, default=0, metavar='N',
|
|
|
|
|
parser.add_argument('--checkpoint-hist', type=int, default=10, metavar='N',
|
|
|
|
|
help='number of checkpoints to keep (default: 10)')
|
|
|
|
|
parser.add_argument('-j', '--workers', type=int, default=4, metavar='N',
|
|
|
|
|
help='how many training processes to use (default: 1)')
|
|
|
|
|
help='how many training processes to use (default: 4)')
|
|
|
|
|
parser.add_argument('--save-images', action='store_true', default=False,
|
|
|
|
|
help='save images of input bathes every log interval for debugging')
|
|
|
|
|
parser.add_argument('--amp', action='store_true', default=False,
|
|
|
|
|