diff --git a/example/alexnet_cifar10/eval.py b/example/alexnet_cifar10/eval.py index 2efc6d15f69b693297872fbbc84ce0f568499ff8..503af7cf75e66ea937130ac7111a1871268d4dd7 100644 --- a/example/alexnet_cifar10/eval.py +++ b/example/alexnet_cifar10/eval.py @@ -36,7 +36,7 @@ if __name__ == "__main__": parser.add_argument('--data_path', type=str, default="./", help='path where the dataset is saved') parser.add_argument('--ckpt_path', type=str, default="./ckpt", help='if is test, must provide\ path where the trained ckpt file') - parser.add_argument('--dataset_sink_mode', type=bool, default=False, help='dataset_sink_mode is False or True') + parser.add_argument('--dataset_sink_mode', type=bool, default=True, help='dataset_sink_mode is False or True') args = parser.parse_args() context.set_context(mode=context.GRAPH_MODE, device_target=args.device_target) diff --git a/example/alexnet_cifar10/train.py b/example/alexnet_cifar10/train.py index 0a288ea1db1297c64e3eb3eb89ef03eb07575a5e..4a3cfdd89f48ffbbea83818957195636a1bbce48 100644 --- a/example/alexnet_cifar10/train.py +++ b/example/alexnet_cifar10/train.py @@ -38,7 +38,7 @@ if __name__ == "__main__": parser.add_argument('--data_path', type=str, default="./", help='path where the dataset is saved') parser.add_argument('--ckpt_path', type=str, default="./ckpt", help='if is test, must provide\ path where the trained ckpt file') - parser.add_argument('--dataset_sink_mode', type=bool, default=False, help='dataset_sink_mode is False or True') + parser.add_argument('--dataset_sink_mode', type=bool, default=True, help='dataset_sink_mode is False or True') args = parser.parse_args() context.set_context(mode=context.GRAPH_MODE, device_target=args.device_target)