Open osuossu8 opened 3 years ago
pip install efficientnet_pytorch
from efficientnet_pytorch import EfficientNet
class feature_extractor(nn.Module):
def __init__(self, original):
super().__init__()
self.model = original
def forward(self, x):
x= self.model.extract_features(x)
return x
class PANNsCNN14Att(nn.Module):
def __init__():
self.Enet = feature_extractor(EfficientNet.from_pretrained('efficientnet-b0'))
...
def cnn_feature_extractor(self, x):
return self.Enet(x)
def forward(self, input, mixup_lambda=None):
x, frames_num = self.preprocess(input, mixup_lambda=mixup_lambda)
x = torch.cat((x,x,x),1)
x = self.cnn_feature_extractor(x)
x = torch.mean(x, dim=3)
...
Q : have you tried any audio augmentations (like Guassian noise etc.)
A : I tried to add white noise and pink noise. But they didn't work.
shinmura0 さんが 0.9 届いた時の学習セット
スレッドタイトル
メモ
ぶらさがってるコメントなど
それらを受けてのアイデア、コメントなど