Allow float decay epochs arg for training, works out with step lr math

5 years ago · 1f39d15f15
parent 7b83e67f77
commit 1f39d15f15
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -89,7 +89,7 @@ parser.add_argument('--epochs', type=int, default=200, metavar='N',
                    help='number of epochs to train (default: 2)')
 parser.add_argument('--start-epoch', default=None, type=int, metavar='N',
                    help='manual epoch number (useful on restarts)')
-parser.add_argument('--decay-epochs', type=int, default=30, metavar='N',
+parser.add_argument('--decay-epochs', type=float, default=30, metavar='N',
                    help='epoch interval to decay LR')
 parser.add_argument('--warmup-epochs', type=int, default=3, metavar='N',
                    help='epochs to warmup LR, if scheduler supports')