from .mlp import MLP
from .net import Net
from .melspectrogram import MelSpectrogram
from .transformer import Transformer
from .conformer import Conformer
from .encodec import EnCodec
from .cqt import CQT
from .waveform import Waveform

NETS = {
    "net": Net,
    "mlp": MLP,
    "melspectrogram": MelSpectrogram,
    "transformer": Transformer,
    "conformer": Conformer,
    "encodec": EnCodec,
    "cqt": CQT,
    "waveform": Waveform,
}
