ModelScope/maas_lib/tools/train.py

# Copyright (c) Alibaba, Inc. and its affiliates.

import argparse

from maas_lib.trainers import build_trainer


def parse_args():
    parser = argparse.ArgumentParser(description='Train a model')
    parser.add_argument('config', help='config file path', type=str)
    parser.add_argument(
        'trainer_name', help='name for trainer', type=str, default=None)
    args = parser.parse_args()
    return args


def main():
    args = parse_args()
    kwargs = dict(cfg_file=args.config)
    trainer = build_trainer(args.trainer_name, kwargs)
    trainer.train()


if __name__ == '__main__':
    main()