Merge pull request #3855 from rerngvit/fix/attention_ocr_momentum_flag

Fix momentum value flag from string => float

Merge pull request #3855 from rerngvit/fix/attention_ocr_momentum_flag
Fix momentum value flag from string => float
047bcef3 · Alexander Gorban · GitHub · 02d40e9f · ccb46317 · 047bcef3
Unverified Commit 047bcef3 authored Apr 03, 2018 by Alexander Gorban Committed by GitHub Apr 03, 2018
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,