| 803 | |
| 804 | # Get parse |
| 805 | def get_arg_parser(): |
| 806 | parser = argparse.ArgumentParser() |
| 807 | parser.add_argument('--data_location', |
| 808 | help='Full path of train data', |
| 809 | required=False, |
| 810 | default='./data') |
| 811 | parser.add_argument('--steps', |
| 812 | help='set the number of steps on train dataset', |
| 813 | type=int, |
| 814 | default=0) |
| 815 | parser.add_argument('--batch_size', |
| 816 | help='Batch size to train. Default is 512', |
| 817 | type=int, |
| 818 | default=2048) |
| 819 | parser.add_argument('--output_dir', |
| 820 | help='Full path to model output directory. \ |
| 821 | Default to ./result. Covered by --checkpoint. ', |
| 822 | required=False, |
| 823 | default='./result') |
| 824 | parser.add_argument('--checkpoint', |
| 825 | help='Full path to checkpoints input/output. \ |
| 826 | Default to ./result/$MODEL_TIMESTAMP', |
| 827 | required=False) |
| 828 | parser.add_argument('--save_steps', |
| 829 | help='set the number of steps on saving checkpoints', |
| 830 | type=int, |
| 831 | default=0) |
| 832 | parser.add_argument('--seed', |
| 833 | help='set the random seed for tensorflow', |
| 834 | type=int, |
| 835 | default=2021) |
| 836 | parser.add_argument('--optimizer', |
| 837 | type=str, \ |
| 838 | choices=['adam', 'adamasync', 'adagraddecay'], |
| 839 | default='adamasync') |
| 840 | parser.add_argument('--learning_rate', |
| 841 | help='Learning rate for model', |
| 842 | type=float, |
| 843 | default=0.001) |
| 844 | parser.add_argument('--keep_checkpoint_max', |
| 845 | help='Maximum number of recent checkpoint to keep', |
| 846 | type=int, |
| 847 | default=1) |
| 848 | parser.add_argument('--timeline', |
| 849 | help='number of steps on saving timeline. Default 0', |
| 850 | type=int, |
| 851 | default=0) |
| 852 | parser.add_argument('--protocol', |
| 853 | type=str, |
| 854 | choices=['grpc', 'grpc++', 'star_server'], |
| 855 | default='grpc') |
| 856 | parser.add_argument('--inter', |
| 857 | help='set inter op parallelism threads.', |
| 858 | type=int, |
| 859 | default=0) |
| 860 | parser.add_argument('--intra', |
| 861 | help='set inter op parallelism threads.', |
| 862 | type=int, |