add tanh to the end of the latent thingy
This commit is contained in:
parent
b7758f25a9
commit
9d7ce42630
|
@ -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):
|
||||||
|
|
Loading…
Reference in New Issue
Block a user