SuperFeatures / how /layers /__init__.py
YannisK's picture
temp state
32408ed
"""
Modules implementing layers in pytorch by inheriting from torch.nn.Module
"""
from . import attention, dim_reduction, pooling