o ߥµiCã@s~ddlmZmZddlmZmZmZddlmZddl m Z ddl m Z e ƒZ edZeeeefefZGdd„deƒZd S) é)ÚABCÚabstractmethod)ÚAnyÚDictÚUnion)ÚModel)Ú ConfigDict)Ú get_logger)z torch.Tensorz tf.Tensorc@sLeZdZdZdd„Zedeeeffdd„ƒZ edeeeffdd„ƒZ d S) ÚHeadzAThe head base class is for the tasks head method definition cKst|ƒ|_dS)N)rÚconfig)ÚselfÚkwargs©rúT/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/models/base/base_head.pyÚ__init__sz Head.__init__ÚreturncOódS)zÕ This method will use the output from backbone model to do any downstream tasks. Receive The output from backbone model. Returns (Dict[str, Any]): The output from downstream task. Nr©r Úargsr rrrÚforwardsz Head.forwardcOr)zo compute loss for head during the finetuning. Returns (Dict[str, Any]): The loss dict NrrrrrÚ compute_loss!szHead.compute_lossN) Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrÚstrrrrrrrrr s r N)ÚabcrrÚtypingrrrÚ!modelscope.models.base.base_modelrÚmodelscope.utils.configrÚmodelscope.utils.loggerr ÚloggerÚTensorrÚInputr rrrrÚs