INNER CODE UNIT · Python

get_tasks_args

openhackathons-org/gpubootcamp · archived/ai/Megatron/English/Python/jupyter_notebook/Megatron-LM/tasks/vision/main.py:34

def get_tasks_args(parser):
    """Provide extra arguments required for tasks."""
    group = parser.add_argument_group(title="tasks")

    group.add_argument(
        "--epochs",
        type=int,
        default=None,
        help="Number of finetunning epochs. Zero results in "
        "evaluation only.",
    )
    group.add_argument(
        "--pretrained-checkpoint",
        type=str,
        default=None,
        help="Pretrained checkpoint used for finetunning.",
    )
    group.add_argument(

View source record →