Add parameters about retrain.
(opt)
| 417 | |
| 418 | |
| 419 | def add_retrain_params(opt): |
| 420 | """ |
| 421 | Add parameters about retrain. |
| 422 | """ |
| 423 | opt.add_argument("--pre_trained", |
| 424 | type=str, |
| 425 | default=None, |
| 426 | help="Pretrained checkpoint path.") |
| 427 | opt.add_argument("--save_checkpoint_path", |
| 428 | type=str, |
| 429 | default=None, |
| 430 | help="Save checkpoint path.") |
| 431 | opt.add_argument("--save_checkpoint_obs_path", |
| 432 | type=str, |
| 433 | default=None, |
| 434 | help="Save checkpoint path on OBS.") |
| 435 | opt.add_argument("--keep_checkpoint_max", |
| 436 | type=int, |
| 437 | default=1, |
| 438 | help="Max checkpoint save number.") |
| 439 | opt.add_argument("--save_checkpoint_steps", |
| 440 | type=int, |
| 441 | default=2000, |
| 442 | help="Save checkpoint step number.") |
| 443 | opt.add_argument("--save_checkpoint", |
| 444 | type=ast.literal_eval, |
| 445 | default=False, |
| 446 | help="Whether save checkpoint in local disk.") |
| 447 | opt.add_argument("--ckpt_name_prefix", |
| 448 | type=str, |
| 449 | default="pangu", |
| 450 | help="Saving checkpoint name prefix.") |
| 451 | opt.add_argument("--has_trained_epoches", |
| 452 | type=int, |
| 453 | default=0, |
| 454 | help="Epoches has been trained before.") |
| 455 | opt.add_argument("--has_trained_steps", |
| 456 | type=int, |
| 457 | default=0, |
| 458 | help="Steps has been trained before.") |
| 459 | |
| 460 | |
| 461 | def get_args(inference=False): |