Train: Lower the initial learning rate to 0.00075 from 0.001

ljanyst · Feb 3, 2019 · a1d02bd · a1d02bd
1 parent ebf1b11
commit a1d02bd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -62,7 +62,7 @@ def main():
                         help='name of the tensorboard data directory')
     parser.add_argument('--checkpoint-interval', type=int, default=5,
                         help='checkpoint interval')
-    parser.add_argument('--lr-values', type=str, default='0.001;0.0001;0.00001',
+    parser.add_argument('--lr-values', type=str, default='0.00075;0.0001;0.00001',
                         help='learning rate values')
     parser.add_argument('--lr-boundaries', type=str, default='320000;400000',
                         help='learning rate chage boundaries (in batches)')