"...csrc/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "591c899c6e9272e5a3c0cd2e3bddf82d5fad645e"
Commit ccb46317 authored by rerngvit's avatar rerngvit
Browse files

Fix momentum value flag from string to float

parent a7aa25d3
...@@ -73,7 +73,7 @@ def define(): ...@@ -73,7 +73,7 @@ def define():
flags.DEFINE_string('optimizer', 'momentum', flags.DEFINE_string('optimizer', 'momentum',
'the optimizer to use') 'the optimizer to use')
flags.DEFINE_string('momentum', 0.9, flags.DEFINE_float('momentum', 0.9,
'momentum value for the momentum optimizer if used') 'momentum value for the momentum optimizer if used')
flags.DEFINE_bool('use_augment_input', True, flags.DEFINE_bool('use_augment_input', True,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment