From b040e12d7691534ae8de24eb737110deb4a3c7eb Mon Sep 17 00:00:00 2001 From: minqiyang Date: Wed, 26 Sep 2018 14:18:22 +0800 Subject: [PATCH] Fix image_classification syntax --- fluid/image_classification/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fluid/image_classification/train.py b/fluid/image_classification/train.py index bfc5f8b1..238c322b 100644 --- a/fluid/image_classification/train.py +++ b/fluid/image_classification/train.py @@ -33,7 +33,7 @@ add_arg('lr', float, 0.1, "set learning rate.") add_arg('lr_strategy', str, "piecewise_decay", "Set the learning rate decay strategy.") add_arg('model', str, "SE_ResNeXt50_32x4d", "Set the network to use.") add_arg('enable_ce', bool, False, "If set True, enable continuous evaluation job.") -add_arg('data_dir' str, "./data/ILSVRC2012", "The ImageNet dataset root dir.") +add_arg('data_dir', str, "./data/ILSVRC2012", "The ImageNet dataset root dir.") # yapf: enable model_list = [m for m in dir(models) if "__" not in m] -- GitLab