From 791d510f1e4fba487757799500c70ca8aa1fb107 Mon Sep 17 00:00:00 2001 From: wukesong Date: Mon, 10 Aug 2020 11:46:16 +0800 Subject: [PATCH] modify dafault ckpt path --- chapter03/lenet/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/chapter03/lenet/main.py b/chapter03/lenet/main.py index 7f3dff9..aec53e3 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') -- GitLab