CoolFace
Apppublic

Linhz/ViMNer

sourceHugging Faceupdated 2y agoView on Hugging Face
1likes
rootModelMultimodelNER__pycache__resnet_utils.cpython-39.pyc
resnet_utils.cpython-39.pyc10 linesDownload Raw Back to __pycache__
1a

2;�df��@sHddlZddlmZddlmZddlmmZGdd�dej�Z	dS)�N)�Variablecs&eZdZ�fdd�Zddd�Z�ZS)�myResnetcs$tt|���||_||_||_dS)N)�superr�__init__�resnet�if_fine_tune�device)�selfrrr��	__class__��?E:\demo_datn\pythonProject1\Model\MultimodelNER\resnet_utils.pyrszmyResnet.__init__�cCs�|j�|�}|j�|�}|j�|�}|j�|�}|j�|�}|j�|�}|j�|�}|j�|�}|�	d��	d�}t3�|||g�}|j�|�}|�
|�d�d�}|js�t|j�}t|j�}t|j�}|||fS)N��r�����)r�conv1�bn1�relu�maxpool�layer1�layer2�layer3�layer4�mean�F�adaptive_avg_pool2d�avgpool�view�sizerr�data)r	�xZatt_size�fcZattrrr
�forwards"456zmyResnet.forward)r)�__name__�7__module__�__qualname__rr#�
__classcell__rrr8r
rsr)9�torch�torch.nn�nn�torch.autogradr�torch.nn.functional�10functionalr�Modulerrrrr
�<module>s