tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Stardust·减
2023-09-03 13:27:24 +08:00
committed by GitHub
parent ffad37c975
commit 6ec8e78d4f
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
attentions.py
View File

@@ -43,11 +43,11 @@ class Encoder(nn.Module):
self.kernel_size = kernel_size
self.p_dropout = p_dropout
self.window_size = window_size
if isflow:
cond_layer = torch.nn.Conv1d(256, 2*hidden_channels*n_layers, 1)
self.cond_pre = torch.nn.Conv1d(hidden_channels, 2*hidden_channels, 1)
self.cond_layer = weight_norm(cond_layer, name='weight')
self.gin_channels = 256
#if isflow:
# cond_layer = torch.nn.Conv1d(256, 2*hidden_channels*n_layers, 1)
# self.cond_pre = torch.nn.Conv1d(hidden_channels, 2*hidden_channels, 1)
# self.cond_layer = weight_norm(cond_layer, name='weight')
# self.gin_channels = 256
self.cond_layer_idx = self.n_layers
if 'gin_channels' in kwargs:
self.gin_channels = kwargs['gin_channels']