Apppublic

xdecoder/Instruct-X-Decoder

sourceHugging Faceafl-3.0updated 3y agoView on Hugging Face

163likes

rootutils__pycache__model.cpython-38.pyc

model.cpython-38.pyc10 linesDownload Raw Back to __pycache__

1U

2E$�c��3@s�ddlZddlZddlZddlZddlZddlmZddlm	Z	ddlm4Z5ddlmZe�e
�Zejjejjejjejjejjejjejjejjejjejjg6Zdd�ZdS)�N)�FlopCountAnalysis)�flop_count_table)�flop_count_strcCst�|�|S)N)�NORM_MODULES�append)�cls�r�$/data/arXiv/demo/Demo/utils/model.py�register_norm_modules7r8)�logging�os�time�pickle�torchZtorch.distributed�distributed�distZ	fvcore.nnrrr�	getLogger�__name__�logger�nn�BatchNorm1d�BatchNorm2d�BatchNorm3d�
SyncBatchNorm�	GroupNorm�InstanceNorm1d�InstanceNorm2d�InstanceNorm3d�	LayerNorm�LocalResponseNormrr9rrrr	�<module>s*10�