From f352204aaa48c92c5fffe09455d217835d847cf6 Mon Sep 17 00:00:00 2001 From: root Date: Thu, 17 Sep 2020 10:08:53 +0800 Subject: [PATCH] transformer ckpt path --- model_zoo/official/nlp/transformer/train.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/model_zoo/official/nlp/transformer/train.py b/model_zoo/official/nlp/transformer/train.py index 37c8b05cc1..5ad68a681c 100644 --- a/model_zoo/official/nlp/transformer/train.py +++ b/model_zoo/official/nlp/transformer/train.py @@ -103,8 +103,7 @@ def argparse_init(): parser.add_argument("--save_checkpoint_steps", type=int, default=2500, help="Save checkpoint steps, " "default is 2500.") parser.add_argument("--save_checkpoint_num", type=int, default=30, help="Save checkpoint numbers, default is 30.") - parser.add_argument("--save_checkpoint_path", type=str, default="./checkpoint/", help="Save checkpoint file path, " - "default is ./checkpoint/") + parser.add_argument("--save_checkpoint_path", type=str, default="./", help="Save checkpoint file path") parser.add_argument("--data_path", type=str, default="", help="Data path, it is better to use absolute path") parser.add_argument("--bucket_boundaries", type=ast.literal_eval, default=[16, 32, 48, 64, 128], help="sequence length for different bucket")