LICENSE
README.md
setup.py
torch_modules_compilation/__init__.py
torch_modules_compilation.egg-info/PKG-INFO
torch_modules_compilation.egg-info/SOURCES.txt
torch_modules_compilation.egg-info/dependency_links.txt
torch_modules_compilation.egg-info/requires.txt
torch_modules_compilation.egg-info/top_level.txt
torch_modules_compilation/modules/BottleneckResBlock.py
torch_modules_compilation/modules/DepthwiseSepConv.py
torch_modules_compilation/modules/FeatureMapSelfAttention.py
torch_modules_compilation/modules/GLAM.py
torch_modules_compilation/modules/GlobalContextModule.py
torch_modules_compilation/modules/LFSATokenizer.py
torch_modules_compilation/modules/ParameterFreeChannelAttention.py
torch_modules_compilation/modules/PatchMerger.py
torch_modules_compilation/modules/RefinementBlock.py
torch_modules_compilation/modules/ResBlock.py
torch_modules_compilation/modules/ResBlockUpDownSample.py
torch_modules_compilation/modules/ResidualMLP_block.py
torch_modules_compilation/modules/ResidualMLP_downsample.py
torch_modules_compilation/modules/SEModule.py
torch_modules_compilation/modules/TokenLearner.py
torch_modules_compilation/modules/TransformerEncoderLayer.py
torch_modules_compilation/modules/TripletAttention.py
torch_modules_compilation/modules/UnetDecoder.py
torch_modules_compilation/modules/UnetEncoder.py
torch_modules_compilation/modules/__init__.py
torch_modules_compilation/modules/global_channel_attention.py
torch_modules_compilation/modules/global_spatial_attention.py
torch_modules_compilation/modules/local_channel_attention.py
torch_modules_compilation/modules/local_spatial_attention.py