From a1d02bd8aa7fb2f26b794605dbf6f3ecede9baf5 Mon Sep 17 00:00:00 2001 From: Lukasz Janyst Date: Sun, 3 Feb 2019 19:26:28 +0100 Subject: [PATCH] Train: Lower the initial learning rate to 0.00075 from 0.001 --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 7789538..2677dda 100755 --- 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)')