We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7b83e67 commit 1f39d15Copy full SHA for 1f39d15
train.py
@@ -89,7 +89,7 @@
89
help='number of epochs to train (default: 2)')
90
parser.add_argument('--start-epoch', default=None, type=int, metavar='N',
91
help='manual epoch number (useful on restarts)')
92
-parser.add_argument('--decay-epochs', type=int, default=30, metavar='N',
+parser.add_argument('--decay-epochs', type=float, default=30, metavar='N',
93
help='epoch interval to decay LR')
94
parser.add_argument('--warmup-epochs', type=int, default=3, metavar='N',
95
help='epochs to warmup LR, if scheduler supports')
0 commit comments