From 725ea3f17d38edeb4648694f6115c0ff9cd2efe4 Mon Sep 17 00:00:00 2001 From: "yi.wu" Date: Tue, 5 Jun 2018 16:01:27 +0800 Subject: [PATCH] update --- benchmark/fluid/models/resnet.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/benchmark/fluid/models/resnet.py b/benchmark/fluid/models/resnet.py index 9ccd41c9e95..47d8d026edc 100644 --- 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 = [ -- GitLab