o rq i%@sddlZddlZddlmZmZddlmZmZmZddl m Z ddl m Z ddl mZddlmZddlmZmZd d lmZGd d d eZejd dGdddeZdS)N)ABCabstractmethod)CallableDictOptional)check_local_model_is_latest)snapshot_download)TRAINERS)Config)Invoke ThirdParty) LogBufferc@s`eZdZdZddedeefddZ  dddZe d d Z e d ed e ee ffd dZ dS) BaseTrainerz Base class for trainer which can not be instantiated. BaseTrainer defines necessary interface and provide default implementation for basic initialization such as parsing config file and parsing commandline args. Ncfg_file arg_parse_fncCsLt||_|r|j||_nd|_t|_t|_t dt |_ dS)z Trainer basic init, should be called in derived class Args: cfg_file: Path to configuration file. arg_parse_fn: Same as ``parse_fn`` in :obj:`Config.to_args`. Nz %Y%m%d_%H%M%S) r from_filecfgto_argsargsrZ log_bufferZvisualization_buffertimestrftime localtime timestamp)selfrrr_/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/modelscope/trainers/base.py__init__s zBaseTrainer.__init__cCsftj|r#tj|r|ntj|}t|tjtjt j|id|St ||tjtj t j|id}|S)aH Get local model directory or download model if necessary. Args: model (str): model id or path to local model directory. model_revision (str, optional): model version number. third_party (str, optional): in which third party library this function is called. ) user_agent)revisionr) ospathexistsisdirdirnamerr ZKEYZ LOCAL_TRAINERr rZTRAINER)rmodelZmodel_revisionZ third_partyZmodel_cache_dirrrrget_or_download_model_dir(s*  z%BaseTrainer.get_or_download_model_dircOdS) Train (and evaluate) process Train process should be implemented for specific task or model, related parameters have been initialized in ``BaseTrainer.__init__`` and should be used in this function Nr)rrkwargsrrrtrainGszBaseTrainer.traincheckpoint_pathreturncOr') Evaluation process Evaluation process should be implemented for specific task or model, related parameters have been initialized in ``BaseTrainer.__init__`` and should be used in this function Nr)rr+rr)rrrevaluateQs zBaseTrainer.evaluateN)NN)__name__ __module__ __qualname____doc__strrrrr&rr*rfloatr.rrrrrs    rdummy) module_namecsHeZdZdeffdd ZddZ d dedeeeffd d ZZ S) DummyTrainerrcst|dS)zY Dummy Trainer. Args: cfg_file: Path to configuration file. N)superr)rrrr) __class__rrr`szDummyTrainer.__init__cOs|jj}td|dS)r(z train cfg N)rr*print)rrr)rrrrr*hszDummyTrainer.trainNr+r,cOs(|jj}td|td|dS)r-z eval cfg zcheckpoint_path N)rZ evaluationr<)rr+rr)rrrrr.rs zDummyTrainer.evaluater/) r0r1r2r4rr*rr5r. __classcell__rrr:rr8]s  r8)r rabcrrtypingrrrZmodelscope.hub.check_modelrZ modelscope.hub.snapshot_downloadrZmodelscope.trainers.builderr Zmodelscope.utils.configr Zmodelscope.utils.constantr r Zutils.log_bufferrrZregister_moduler8rrrrs      M