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

add tanh to the end of the latent thingy

Browse Source
This commit is contained in:
James Betker 2022-06-16 20:31:23 -06:00
parent b7758f25a9
commit 9d7ce42630
1 changed files with 2 additions and 1 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

@ -35,7 +35,8 @@ class UpperEncoder(nn.Module):
ResBlock(hidden_dim, out_channels=hidden_dim, use_conv=True, dims=1), ResBlock(hidden_dim, out_channels=hidden_dim, use_conv=True, dims=1),
nn.GroupNorm(8, hidden_dim), nn.GroupNorm(8, hidden_dim),
nn.SiLU(), nn.SiLU(),
nn.Conv1d(hidden_dim, embedding_dim, 1) nn.Conv1d(hidden_dim, embedding_dim, 1),
nn.Tanh(),
) )
def forward(self, x): def forward(self, x):