fix tf.app.flags.DEFINE_float

satojkovic · Apr 14, 2018 · 6612567 · 6612567
1 parent 964eb9a
commit 6612567
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_deep_logo_cnn.py b/train_deep_logo_cnn.py
@@ -42,7 +42,7 @@
                             "A width of an input image.")
 tf.app.flags.DEFINE_integer("image_height", common.CNN_IN_HEIGHT,
                             "A height of an input image.")
-tf.app.flags.DEFINE_integer("learning_rate", 0.0001, "Learning rate")
+tf.app.flags.DEFINE_float("learning_rate", 0.0001, "Learning rate")
 tf.app.flags.DEFINE_integer("batch_size", 64, "A batch size")
 tf.app.flags.DEFINE_integer("num_channels", common.CNN_IN_CH,
                             "A number of channels of an input image.")