Merge pull request #3 from wjbianjason/patch-3

update param
zhougr1993 · Aug 30, 2017 · b820cc4 · b820cc4
2 parents baeafd3 + ac3bad4
commit b820cc4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rocket_bottom.py b/rocket_bottom.py
@@ -53,7 +53,7 @@
 parser.add_argument('--epochs', default=200, type=int, metavar='N',
                     help='number of total epochs to run')
 parser.add_argument('--weightDecay', default=0.0005, type=float)
-parser.add_argument('--epoch_step', default='[60,120,160]', type=str,
+parser.add_argument('--epoch_step', default='[70,130,190]', type=str,
                     help='json list with epochs to drop lr on')
 parser.add_argument('--lr_decay_ratio', default=0.2, type=float)
 parser.add_argument('--resume', default='', type=str)