diff --git a/train.py b/train.py index 4264a164..16c4b051 100755 --- a/train.py +++ b/train.py @@ -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,