Browse Source

transformer ckpt path

tags/v1.0.0
root 5 years ago
parent
commit
f352204aaa
1 changed files with 1 additions and 2 deletions
  1. +1
    -2
      model_zoo/official/nlp/transformer/train.py

+ 1
- 2
model_zoo/official/nlp/transformer/train.py View File

@@ -103,8 +103,7 @@ def argparse_init():
parser.add_argument("--save_checkpoint_steps", type=int, default=2500, help="Save checkpoint steps, " parser.add_argument("--save_checkpoint_steps", type=int, default=2500, help="Save checkpoint steps, "
"default is 2500.") "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_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("--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], parser.add_argument("--bucket_boundaries", type=ast.literal_eval, default=[16, 32, 48, 64, 128],
help="sequence length for different bucket") help="sequence length for different bucket")


Loading…
Cancel
Save