| 709 | |
| 710 | # Get parse |
| 711 | def get_arg_parser(): |
| 712 | parser = argparse.ArgumentParser() |
| 713 | parser.add_argument('--data_location', |
| 714 | help='Full path of train data', |
| 715 | required=False, |
| 716 | default='./data') |
| 717 | parser.add_argument('--steps', |
| 718 | help='set the number of steps on train dataset', |
| 719 | type=int, |
| 720 | default=0) |
| 721 | parser.add_argument('--batch_size', |
| 722 | help='Batch size to train. Default is 512', |
| 723 | type=int, |
| 724 | default=2048) |
| 725 | parser.add_argument('--output_dir', |
| 726 | help='Full path to logs & model output directory', |
| 727 | required=False, |
| 728 | default='./result') |
| 729 | parser.add_argument('--checkpoint', |
| 730 | help='Full path to checkpoints input/output directory', |
| 731 | required=False) |
| 732 | parser.add_argument('--deep_dropout', |
| 733 | help='Dropout regularization for deep model', |
| 734 | type=float, |
| 735 | default=0.0) |
| 736 | parser.add_argument('--seed', |
| 737 | help='set the random seed for tensorflow', |
| 738 | type=int, |
| 739 | default=2021) |
| 740 | parser.add_argument('--optimizer', |
| 741 | type=str, |
| 742 | choices=['adam', 'adamasync', 'adagraddecay', |
| 743 | 'adagrad', 'gradientdescent'], |
| 744 | default='adamasync') |
| 745 | parser.add_argument('--learning_rate', |
| 746 | help='Learning rate for model', |
| 747 | type=float, |
| 748 | default=0.01) |
| 749 | parser.add_argument('--save_steps', |
| 750 | help='set the number of steps on saving checkpoints', |
| 751 | type=int, |
| 752 | default=0) |
| 753 | parser.add_argument('--keep_checkpoint_max', |
| 754 | help='Maximum number of recent checkpoint to keep', |
| 755 | type=int, |
| 756 | default=1) |
| 757 | parser.add_argument('--bf16', |
| 758 | help='enable DeepRec BF16 in deep model. Default FP32', |
| 759 | action='store_true') |
| 760 | parser.add_argument('--no_eval', |
| 761 | help='not evaluate trained model by eval dataset.', |
| 762 | action='store_true') |
| 763 | parser.add_argument('--timeline', |
| 764 | help='number of steps on saving timeline. Default 0', |
| 765 | type=int, |
| 766 | default=0) |
| 767 | parser.add_argument('--interaction_op', |
| 768 | type=str, |