mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Reduce severity of gpt_music reduction

Browse Source
This commit is contained in:
James Betker 2022-06-16 20:05:05 -06:00
parent 41abf776d9
commit e3287c9d95
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
codes/models/audio/music/gpt_music2.py
View File

@ -24,9 +24,6 @@ class UpperEncoder(nn.Module):
ResBlock(spec_dim, out_channels=edim(1), use_conv=True, dims=1, down=True),
ResBlock(edim(1), out_channels=edim(2), use_conv=True, dims=1, down=True),
ResBlock(edim(2), out_channels=edim(3), use_conv=True, dims=1, down=True),
ResBlock(edim(3), out_channels=edim(4), use_conv=True, dims=1, down=True),
ResBlock(edim(4), out_channels=edim(4), use_conv=True, dims=1),
ResBlock(edim(4), out_channels=edim(5), use_conv=True, dims=1, down=True),
ResBlock(edim(5), out_channels=edim(5), use_conv=True, dims=1),
ResBlock(edim(5), out_channels=hidden_dim, use_conv=True, dims=1, down=True))
self.encoder = nn.Sequential(