Closed xinmokunshou closed 3 years ago
` class Transfer_Cnn14(nn.Module): | def init(self, sample_rate, window_size, hop_size, mel_bins, fmin, | fmax, classes_num, freeze_base): | """Classifier for a new task using pretrained Cnn14 as a sub module. | """ | super(Transfer_Cnn14, self).init() | audioset_classes_num = 527 | | self.base = Cnn14(sample_rate, window_size, hop_size, mel_bins, fmin, | fmax, audioset_classes_num) | | # Transfer to another task layer | self.fc_transfer = nn.Linear(2048, classes_num, bias=True)
` GRZAN datasets contains 10 classes，and in the model, classes_num=527, what do i have to do to make this work?
` class Transfer_Cnn14(nn.Module): | def init(self, sample_rate, window_size, hop_size, mel_bins, fmin, | fmax, classes_num, freeze_base): | """Classifier for a new task using pretrained Cnn14 as a sub module. | """ | super(Transfer_Cnn14, self).init() | audioset_classes_num = 527 | | self.base = Cnn14(sample_rate, window_size, hop_size, mel_bins, fmin, | fmax, audioset_classes_num) | | # Transfer to another task layer | self.fc_transfer = nn.Linear(2048, classes_num, bias=True)
` GRZAN datasets contains 10 classes，and in the model, classes_num=527, what do i have to do to make this work?