|
|
|
@@ -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") |
|
|
|
|