diff --git a/chapter03/lenet/main.py b/chapter03/lenet/main.py index 7f3dff9104eb0396b085a917779c8796301c2f9c..aec53e3322b0940ca8984eba4cfd7b1863c69b9c 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')