From 661256737432d56e4a60386e1218ca0d932be81e Mon Sep 17 00:00:00 2001 From: Yury Belousov Date: Sat, 14 Apr 2018 14:10:06 +0300 Subject: [PATCH] fix tf.app.flags.DEFINE_float --- train_deep_logo_cnn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train_deep_logo_cnn.py b/train_deep_logo_cnn.py index c95a538..397c894 100644 --- 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.")