Mambaをdcase2024に移植する

MambaをDCASE 2024のベースラインで使う

RNNモジュールを置き換える

移植を考える

RNNモジュールがCRNN

RNN関連の設定

code: _

rnn_layers: 1

n_in_channel: 1

n_RNN_cell: 192

activation: glu

rnn_type: BGRU

実装 >

CRNN中の実装

code: init.py

if rnn_type == "BGRU":

nb_in = self.cnn.nb_filters-1

if self.cnn_integration:

# self.fc = nn.Linear(nb_in * n_in_channel, nb_in)

nb_in = nb_in * n_in_channel

self.rnn = BidirectionalGRU(

n_in=nb_in,

n_hidden=n_RNN_cell,

dropout=dropout_recurrent,

num_layers=n_layers_RNN,

)

else:

NotImplementedError("Only BGRU supported for CRNN for now")

呼ばれるのはfoward内の最後の方

Mambaの実装 >