Skip to content

Commit 1f39d15

Browse files
committed
Allow float decay epochs arg for training, works out with step lr math
1 parent 7b83e67 commit 1f39d15

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@
8989
help='number of epochs to train (default: 2)')
9090
parser.add_argument('--start-epoch', default=None, type=int, metavar='N',
9191
help='manual epoch number (useful on restarts)')
92-
parser.add_argument('--decay-epochs', type=int, default=30, metavar='N',
92+
parser.add_argument('--decay-epochs', type=float, default=30, metavar='N',
9393
help='epoch interval to decay LR')
9494
parser.add_argument('--warmup-epochs', type=int, default=3, metavar='N',
9595
help='epochs to warmup LR, if scheduler supports')

0 commit comments

Comments
 (0)