update

7 years ago · 725ea3f17d
parent 2e0d851685
commit 725ea3f17d
1 changed files with 4 additions and 4 deletions
--- a/benchmark/fluid/models/resnet.py
+++ b/benchmark/fluid/models/resnet.py
@ -142,11 +142,11 @@ def get_model(args):
        else:
            dshape = [224, 224, 3]
        model = resnet_imagenet
-        if not args.data_dir:
+        if not args.data_path:
            raise Exception(
-                "Must specify --data_dir when training with imagenet")
-        train_reader = imagenet_train(args.data_dir)
-        test_reader = imagenet_test(args.data_dir)
+                "Must specify --data_path when training with imagenet")
+        train_reader = imagenet_train(args.data_path)
+        test_reader = imagenet_test(args.data_path)

    if args.use_reader_op:
        filelist = [