multipitch-architectures / libdl / nn_models / __init__.py
__init__.py
Raw
from .basic_cnns import basic_cnn, basic_cnn_pool
from .basic_cnns import basic_cnn_segm_sigmoid, basic_cnn_segm_logsoftmax, basic_cnn_segm_blank_logsoftmax
from .basic_cnns import deep_cnn_segm_sigmoid
from .unet_cnns import single_conv, double_conv, unet_up_concat_padding, transformer_enc_layer, simple_u_net, simple_u_net_largekernels, simple_u_net_selfattn, simple_u_net_doubleselfattn
from .unet_cnns import freq_u_net, freq_u_net_bottomstack, freq_u_net_selfattn, freq_u_net_doubleselfattn
from .unet_cnns import simple_u_net_doubleselfattn_twolayers, simple_u_net_doubleselfattn_alllayers, simple_u_net_doubleselfattn_varlayers, simple_u_net_sixselfattn
from .unet_cnns import u_net_temporal_selfattn_varlayers, transformer_temporal_enc_layer, simple_u_net_doubleselfattn_transenc
from .unet_cnns import blstm_temporal_enc_layer, u_net_blstm_varlayers, u_net_temporal_blstm_varlayers
from .unet_cnns import simple_u_net_doubleselfattn_polyphony, simple_u_net_doubleselfattn_polyphony_classif
from .unet_cnns import simple_u_net_polyphony_classif, simple_u_net_polyphony_classif_softmax