diff --git a/chapter03/lenet/main.py b/chapter03/lenet/main.py index b94200e3a93acb2a0d1fa826678ab670bcce1cf6..1dc82379a886bdf968e9641bc0e106ee5d1dbac6 100644 --- a/chapter03/lenet/main.py +++ b/chapter03/lenet/main.py @@ -82,7 +82,7 @@ if __name__ == "__main__": help='implement phase, set to train or test') parser.add_argument('--data_path', type=str, default="./MNIST_Data", help='path where the dataset is saved') - parser.add_argument('--ckpt_path', type=str, default="", help='if mode is test, must provide\ + parser.add_argument('--ckpt_path', type=str, default="./", help='if mode 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')