27 lines
699 B
Python
27 lines
699 B
Python
# These codes are copied from modelscope revision c58451baead80d83281f063d12fb377fad415257
|
|
# Copyright (c) Alibaba, Inc. and its affiliates.
|
|
|
|
import argparse
|
|
|
|
from modelscope.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()
|