| 721 | return low_string == 'true' |
| 722 | |
| 723 | def get_arg_parser(): |
| 724 | parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter) |
| 725 | parser.add_argument('--data_location', |
| 726 | help='Full path of train data', |
| 727 | required=False, |
| 728 | default='./data') |
| 729 | parser.add_argument('--steps', |
| 730 | help='set the number of steps on train dataset', |
| 731 | type=int, |
| 732 | default=0) |
| 733 | parser.add_argument('--batch_size', |
| 734 | help='Batch size to train', |
| 735 | type=int, |
| 736 | default=2048) |
| 737 | parser.add_argument('--output_dir', |
| 738 | help='Full path to logs & model output directory', |
| 739 | required=False, |
| 740 | default='./result') |
| 741 | parser.add_argument('--checkpoint', |
| 742 | help='Full path to checkpoints input/output directory', |
| 743 | required=False) |
| 744 | parser.add_argument('--model_dir', |
| 745 | help='Full path to test model directory', |
| 746 | required=False) |
| 747 | parser.add_argument('--learning_rate', |
| 748 | help='Learning rate for model', |
| 749 | type=float, |
| 750 | default=0.1) |
| 751 | parser.add_argument('--l2_regularization', |
| 752 | help='L2 regularization for the model', |
| 753 | type=float, |
| 754 | default=L2_REGULARIZATION) |
| 755 | parser.add_argument('--timeline', |
| 756 | help='number of steps on saving timeline', |
| 757 | type=int, |
| 758 | default=0) |
| 759 | parser.add_argument('--save_steps', |
| 760 | help='set the number of steps on saving checkpoints', |
| 761 | type=int, |
| 762 | default=0) |
| 763 | parser.add_argument('--seed', |
| 764 | help='set random seed', |
| 765 | type=int, |
| 766 | default=2021) |
| 767 | parser.add_argument('--keep_checkpoint_max', |
| 768 | help='Maximum number of recent checkpoint to keep', |
| 769 | type=int, |
| 770 | default=1) |
| 771 | parser.add_argument('--bf16', |
| 772 | help='enable DeepRec BF16 in deep model', |
| 773 | action='store_true') |
| 774 | parser.add_argument('--no_eval', |
| 775 | help='not evaluate trained model by eval dataset.', |
| 776 | action='store_true') |
| 777 | parser.add_argument('--protocol', |
| 778 | type=str, |
| 779 | choices=['grpc', 'grpc++', 'star_server'], |
| 780 | default='grpc') |