提交 725ea3f1 编写于 作者: Y yi.wu

update

上级 2e0d8516
...@@ -142,11 +142,11 @@ def get_model(args): ...@@ -142,11 +142,11 @@ def get_model(args):
else: else:
dshape = [224, 224, 3] dshape = [224, 224, 3]
model = resnet_imagenet model = resnet_imagenet
if not args.data_dir: if not args.data_path:
raise Exception( raise Exception(
"Must specify --data_dir when training with imagenet") "Must specify --data_path when training with imagenet")
train_reader = imagenet_train(args.data_dir) train_reader = imagenet_train(args.data_path)
test_reader = imagenet_test(args.data_dir) test_reader = imagenet_test(args.data_path)
if args.use_reader_op: if args.use_reader_op:
filelist = [ filelist = [
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册