From 26029ec95bb0e9fdc637c2d1be14ed24749cde13 Mon Sep 17 00:00:00 2001 From: gengdongjie Date: Sun, 26 Apr 2020 11:41:44 +0800 Subject: [PATCH] Remove default dataset path in cifar_resnet50 tutorial --- tutorials/tutorial_code/resnet/cifar_resnet50.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tutorials/tutorial_code/resnet/cifar_resnet50.py b/tutorials/tutorial_code/resnet/cifar_resnet50.py index 7da052a8..58d2fff0 100644 --- a/tutorials/tutorial_code/resnet/cifar_resnet50.py +++ b/tutorials/tutorial_code/resnet/cifar_resnet50.py @@ -46,7 +46,7 @@ parser.add_argument('--epoch_size', type=int, default=1, help='Epoch size.') parser.add_argument('--batch_size', type=int, default=32, help='Batch size.') parser.add_argument('--num_classes', type=int, default=10, help='Num classes.') parser.add_argument('--checkpoint_path', type=str, default=None, help='CheckPoint file path.') -parser.add_argument('--dataset_path', type=str, default="/var/log/npu/datasets/cifar/cifar-10-batches-bin", help='Dataset path.') +parser.add_argument('--dataset_path', type=str, default=None, help='Dataset path.') args_opt = parser.parse_args() device_id = int(os.getenv('DEVICE_ID')) -- GitLab