Fix momentum value flag from string to float

ccb46317 · rerngvit · a7aa25d3 · ccb46317
Commit ccb46317 authored Apr 03, 2018 by rerngvit
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

research/attention_ocr/python/common_flags.py research/attention_ocr/python/common_flags.py +1 -1

No files found.
--- a/research/attention_ocr/python/common_flags.py
+++ b/research/attention_ocr/python/common_flags.py
@@ -73,7 +73,7 @@ def define():
  flags.DEFINE_string('optimizer', 'momentum',
                      'the optimizer to use')

-  flags.DEFINE_string('momentum', 0.9,
+  flags.DEFINE_float('momentum', 0.9,
                      'momentum value for the momentum optimizer if used')

  flags.DEFINE_bool('use_augment_input', True,