Function get_args

codegeex/mindspore/src/utils.py:461–584

train function for PanguAlpha

(inference=False)

def get_args(inference=False):
    """train function for PanguAlpha"""
    parser = argparse.ArgumentParser(description="PanguAlpha training")
    parser.add_argument('--device_id',
                        type=int,
                        default=0,
                        help="Device id, default is 0.")
    parser.add_argument("--device_num",
                        type=int,
                        default=128,
                        help="Use device nums, default is 128.")
    parser.add_argument("--distribute",
                        type=str,
                        default="true",
                        choices=["true", "false"],
                        help="Run distribute, default is true.")
    parser.add_argument("--load_ckpt_name",
                        type=str,
                        default=None,
                        help="checkpint file name.")
    parser.add_argument("--load_ckpt_path",
                        type=str,
                        default=None,
                        help="checkpoint file path.")
    parser.add_argument("--load_ckpt_epoch",
                        type=int,
                        default=None,
                        help="checkpoint epoch.")
    parser.add_argument('--code_data',
                        type=str,
                        required=True,
                        help='Location of code data.')
    parser.add_argument("--tb_dir",
                        type=str,
                        required=True,
                        help="Location of tensorboard log")
    parser.add_argument("--language",
                        type=str,
                        default=None,
                        help="Language of task")
    parser.add_argument("--part",
                        type=int,
                        default=None,
                        help="Part of task")
    parser.add_argument('--eval_data_url',
                        required=False,
                        default=None,
                        help='Location of eval data.')
    parser.add_argument('--train_url',
                        required=False,
                        default=None,
                        help='Location of training outputs.')
    parser.add_argument("--run_type",
                        type=str,
                        default="predict",
                        choices=["train", "predict"],
                        help="The run type")
    parser.add_argument("--mode",