| 636 | |
| 637 | # Get parse |
| 638 | def get_arg_parser(): |
| 639 | parser = argparse.ArgumentParser() |
| 640 | parser.add_argument('--data_location', |
| 641 | help='Full path of train data', |
| 642 | required=False, |
| 643 | default='./data') |
| 644 | parser.add_argument('--steps', |
| 645 | help='set the number of steps on train dataset', |
| 646 | type=int, |
| 647 | default=0) |
| 648 | parser.add_argument('--batch_size', |
| 649 | help='Batch size to train. Default is 512', |
| 650 | type=int, |
| 651 | default=2048) |
| 652 | parser.add_argument('--output_dir', |
| 653 | help='Full path to model output directory. \ |
| 654 | Default to ./result. Covered by --checkpoint. ', |
| 655 | required=False, |
| 656 | default='./result') |
| 657 | parser.add_argument('--checkpoint', |
| 658 | help='Full path to checkpoints input/output. \ |
| 659 | Default to ./result/$MODEL_TIMESTAMP', |
| 660 | required=False) |
| 661 | parser.add_argument('--save_steps', |
| 662 | help='set the number of steps on saving checkpoints', |
| 663 | type=int, |
| 664 | default=0) |
| 665 | parser.add_argument('--seed', |
| 666 | help='set the random seed for tensorflow', |
| 667 | type=int, |
| 668 | default=2021) |
| 669 | parser.add_argument('--optimizer', |
| 670 | type=str, |
| 671 | choices=['adam', 'adamasync', |
| 672 | 'adagraddecay', 'adagrad'], |
| 673 | default='adamasync') |
| 674 | parser.add_argument('--learning_rate', |
| 675 | help='Learning rate for deep model', |
| 676 | type=float, |
| 677 | default=0.001) |
| 678 | parser.add_argument('--keep_checkpoint_max', |
| 679 | help='Maximum number of recent checkpoint to keep', |
| 680 | type=int, |
| 681 | default=1) |
| 682 | parser.add_argument('--timeline', |
| 683 | help='number of steps on saving timeline. Default 0', |
| 684 | type=int, |
| 685 | default=0) |
| 686 | parser.add_argument('--protocol', |
| 687 | type=str, |
| 688 | choices=['grpc', 'grpc++', 'star_server'], |
| 689 | default='grpc') |
| 690 | parser.add_argument('--inter', |
| 691 | help='set inter op parallelism threads.', |
| 692 | type=int, |
| 693 | default=0) |
| 694 | parser.add_argument('--intra', |
| 695 | help='set inter op parallelism threads.', |