and mel_head

2022-05-06 00:25:27 -06:00 · 2022-05-06 00:25:27 -06:00 · b13d983c24
commit b13d983c24
parent d5fb79564a
1 changed files with 0 additions and 1 deletions
--- a/codes/models/audio/music/music_gen_fill_gaps.py
+++ b/codes/models/audio/music/music_gen_fill_gaps.py
@ -151,7 +151,6 @@ class MusicGenerator(nn.Module):
            DiffusionLayer(model_channels, dropout, num_heads),
        )
        self.integrating_conv = nn.Conv1d(model_channels*2, model_channels, kernel_size=1)
-        self.mel_head = nn.Conv1d(model_channels, in_channels, kernel_size=3, padding=1)
        self.layers = nn.ModuleList([DiffusionLayer(model_channels, dropout, num_heads) for _ in range(num_layers)] +
                                    [ResBlock(model_channels, model_channels, dropout, dims=1, use_scale_shift_norm=True) for _ in range(3)])