o ߥi @s6ddlZddlmZddlmZGdddejZdS)N) ModelFilecseZdZdZeZeddZeddZ eddZ fdd Z d d Z d d Z ddZddZddZ   dddZ    dddZZS)SpaceModelBasez; Basic model wrapper for static graph and dygrpah. cCs|tj|<dSNr _registry)clsnamer `/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/models/nlp/space/model/model_base.pyregisters zSpaceModelBase.registercCs tj|Srr)rr r r by_names zSpaceModelBase.by_namecOs&t|jj}|||g|Ri|Sr)rr Modelmodel) model_dirconfigargskwargs model_clsr r r createszSpaceModelBase.createcs@tt|tj|tj|_|j j |_ |j |_ |j j |_ dSr)superr__init__ospathjoinrTORCH_MODEL_BIN_FILEinit_checkpointDataset abandon_labeluse_gpuTrainergpu)selfrr __class__r r rs  zSpaceModelBase.__init__cCt)z Create model's parameters. NotImplementedError)r!r 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%)r!r)outputsr+ data_filer r r _collect_metrics/r(zSpaceModelBase._collect_metricscCr$)z* Optimize loss function and update model. r%)r!loss optimizer lr_schedulerr r r _optimize3r(zSpaceModelBase._optimizecCr$)z" Real inference process of model. r%)r!r)start_ideos_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.)trainevalrr,r/)r!r)r*r+r.r-metricsr r r forward;s zSpaceModelBase.forwardcCs ||j|||||d}|S)zqInference process. Args: inputs(`dict` of numpy.ndarray/int/float/...) : input data )r4r5r6r7)r:r8)r!r)r4r5r6r7resultsr r r inferRs zSpaceModelBase.infer)FFN)NNNN)__name__ __module__ __qualname____doc__dictr classmethodr staticmethodr rrr'r,r/r3r8r<r> __classcell__r r r"r r s0      r)rtorch.nnnnmodelscope.utils.constantrModulerr r r r s