Change type of 'lr_refactor_ratio' to 'float' or else the learning-ra…

…te-refactor will never work (apache#8982)
zhreshold · Dec 14, 2017 · cb82472 · cb82472
1 parent dfe4717
commit cb82472
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/example/ssd/train.py b/example/ssd/train.py
@@ -72,7 +72,7 @@ def parse_args():
                         help='blue mean value')
     parser.add_argument('--lr-steps', dest='lr_refactor_step', type=str, default='80, 160',
                         help='refactor learning rate at specified epochs')
-    parser.add_argument('--lr-factor', dest='lr_refactor_ratio', type=str, default=0.1,
+    parser.add_argument('--lr-factor', dest='lr_refactor_ratio', type=float, default=0.1,
                         help='ratio to refactor learning rate')
     parser.add_argument('--freeze', dest='freeze_pattern', type=str, default="^(conv1_|conv2_).*",
                         help='freeze layer pattern')