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.kernel_size = kernel_size
self.p_dropout = p_dropout self.p_dropout = p_dropout
self.window_size = window_size self.window_size = window_size
if isflow: #if isflow:
cond_layer = torch.nn.Conv1d(256, 2*hidden_channels*n_layers, 1) # 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_pre = torch.nn.Conv1d(hidden_channels, 2*hidden_channels, 1)
self.cond_layer = weight_norm(cond_layer, name='weight') # self.cond_layer = weight_norm(cond_layer, name='weight')
self.gin_channels = 256 # self.gin_channels = 256
self.cond_layer_idx = self.n_layers self.cond_layer_idx = self.n_layers
if 'gin_channels' in kwargs: if 'gin_channels' in kwargs:
self.gin_channels = kwargs['gin_channels'] self.gin_channels = kwargs['gin_channels']