o
    ß¥µi]  ã                   @   sP   d dl mZmZ d dlZd dlmZ d dlmZ eƒ ZG dd„ deej	j
ƒZdS )é    )ÚAnyÚDictN)ÚHead)Ú
get_loggerc                       sL   e Zd ZdZ‡ fdd„Zdeeef fdd„Zdeeef fdd„Z	‡  Z
S )	Ú	TorchHeadz& Base head interface for pytorch

    c                    s$   t ƒ jdi |¤Ž tjj | ¡ d S )N© )ÚsuperÚ__init__ÚtorchÚnnÚModule)ÚselfÚkwargs©Ú	__class__r   úZ/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/models/base/base_torch_head.pyr	      s   zTorchHead.__init__Úreturnc                 O   ó   t ‚©N©ÚNotImplementedError©r   Úargsr   r   r   r   Úforward   ó   zTorchHead.forwardc                 O   r   r   r   r   r   r   r   Úcompute_loss   r   zTorchHead.compute_loss)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r	   r   Ústrr   r   r   Ú__classcell__r   r   r   r   r      s
    r   )Útypingr   r   r
   Ú modelscope.models.base.base_headr   Úmodelscope.utils.loggerr   Úloggerr   r   r   r   r   r   r   Ú<module>   s   