o
    ॵi_                     @   s:   d dl Z d dlmZ dd Zdd Zedkre  dS dS )    N)build_trainerc                  C   s:   t jdd} | jddtd | jddtd d |  }|S )	NzTrain a model)descriptionconfigzconfig file path)helptypetrainer_namezname for trainer)r   r   default)argparseArgumentParseradd_argumentstr
parse_args)parserargs r   J/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/tools/train.pyr      s   r   c                  C   s*   t  } t| jd}t| j|}|  d S )N)cfg_file)r   dictr   r   r   train)r   kwargstrainerr   r   r   main   s   r   __main__)r	   modelscope.trainersr   r   r   __name__r   r   r   r   <module>   s   	
