o *ÎjØ ã@s6ddlZddlmZddlmZGdd„dejƒZdS)éN)Ú ModelFilecs”eZdZdZeƒZedd„ƒZedd„ƒZ edd„ƒZ ‡fdd „Z d d „Z d d „Z dd„Zdd„Zdd„Z   ddd„Z    ddd„Z‡ZS)ÚSpaceModelBasez; Basic model wrapper for static graph and dygrpah. cCs|tj|<dS©N©rÚ _registry)ÚclsÚname©r úm/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/modelscope/models/nlp/space/model/model_base.pyÚregisters zSpaceModelBase.registercCs tj|Srr)rr r r Úby_names zSpaceModelBase.by_namecOs&t |jj¡}|||g|¢Ri|¤ŽSr)rr ZModelÚmodel)Ú model_dirÚconfigÚargsÚkwargsZ model_clsr r r ÚcreateszSpaceModelBase.createcs@tt|ƒ ¡tj |tj¡|_|j j |_ |j |_ |j j |_ dSr)ÚsuperrÚ__init__ÚosÚpathÚjoinrZTORCH_MODEL_BIN_FILEZinit_checkpointZDatasetÚ abandon_labelZuse_gpuZTrainerZgpu)Úselfrr©Ú __class__r r rsÿ  zSpaceModelBase.__init__cCót‚)z Create model's parameters. ©ÚNotImplementedError)rr r r Ú_create_parameters'óz!SpaceModelBase._create_parameterscCr)zH NO LABEL: Real forward process of model in different mode(train/test). r)rÚinputsÚ is_trainingÚ with_labelr r r Ú_forward+r zSpaceModelBase._forwardcCr)z@ NO LABEL: Calculate loss function by using inputs and outputs. r)rr!Úoutputsr#Ú data_filer r r Ú_collect_metrics/r zSpaceModelBase._collect_metricscCr)z* Optimize loss function and update model. r)rZlossZ optimizerZ lr_schedulerr r r Ú _optimize3r zSpaceModelBase._optimizecCr)z" Real inference process of model. r)rr!Ústart_idÚeos_idÚ max_gen_lenÚ prev_inputr r r Ú_infer7r zSpaceModelBase._inferFNcCsJ|r| ¡n| ¡|jrdn|}|j|||d}|j||||d}|S)z¶ Forward process, include real forward, collect metrices and optimize(optional) Args: inputs(`dict` of numpy.ndarray/int/float/...) : input data F)r#)r#r&)ÚtrainÚevalrr$r')rr!r"r#r&r%Zmetricsr r r Úforward;s ÿzSpaceModelBase.forwardcCs | ¡|j|||||d}|S)zqInference process. Args: inputs(`dict` of numpy.ndarray/int/float/...) : input data )r)r*r+r,)r/r-)rr!r)r*r+r,Úresultsr r r ÚinferRs ûzSpaceModelBase.infer)FFN)NNNN)Ú__name__Ú __module__Ú __qualname__Ú__doc__ÚdictrÚ classmethodr Ú staticmethodr rrrr$r'r(r-r0r2Ú __classcell__r r rr r s0      üûr)rZtorch.nnÚnnZmodelscope.utils.constantrÚModulerr r r r Ús