CoolFace
Modelpublic

RuiTerrty/RemoteSensingChangeDetection-RSCD.HA2F

sourceHugging Faceupdated 6mo agoView on Hugging Face
0likes
rootmodellayers__pycache__mlp.cpython-39.pyc
mlp.cpython-39.pyc12 linesDownload Raw Back to __pycache__
1a

28��g��@s6ddlmZmZddlmZmZGdd�dej�ZdS)�)�Callable�Optional)�Tensor�nnc	s^eZdZddejddfeeeeeedejfe	e3dd��fdd�
Zeed�d	d4�Z
�ZS)�MlpNgT.)�in_features�hidden_features�out_features�	act_layer�drop�bias�returncsVt���|p|}|p|}tj|||d�|_|�|_tj|||d�|_t�|�|_dS)N)r)	�super�__init__r�Linear�fc1�act�fc2�Dropoutr)�selfrrr	r5rr��	__class__��@C:\Users\82607\Desktop\code\ChangeViT-master\model\layers\mlp.pyrs	6zMlp.__init__)�xr
cCs6|�|�}|�|�}|�|�}|�|�}|�|�}|S)N)rrrr)rrrrr�forward"s7891011zMlp.forward)�__name__�12__module__�__qualname__r�GELU�intrr�Module�float�boolrrr�
__classcell__rrrrrs��rN)�typingrr�torchrrr!rrrrr�<module>s