EDGAhab commited on
Commit
8e8e564
·
1 Parent(s): 1c7f273

Update models.py

Browse files
Files changed (1) hide show
  1. models.py +1 -1
models.py CHANGED
@@ -107,7 +107,7 @@ class DurationPredictor(nn.Module):
107
 
108
  self.drop = nn.Dropout(p_dropout)
109
  self.conv_1 = nn.Conv1d(in_channels, filter_channels, kernel_size, padding=kernel_size//2)
110
- self.norm_1 = modules.LayerNorm(filter_channels)
111
  self.conv_2 = nn.Conv1d(filter_channels, filter_channels, kernel_size, padding=kernel_size//2)
112
  self.norm_2 = modules.LayerNorm(filter_channels)
113
  self.proj = nn.Conv1d(filter_channels, 1, 1)
 
107
 
108
  self.drop = nn.Dropout(p_dropout)
109
  self.conv_1 = nn.Conv1d(in_channels, filter_channels, kernel_size, padding=kernel_size//2)
110
+ self.norm_1 = modules.LayerNorm(filter_channels)
111
  self.conv_2 = nn.Conv1d(filter_channels, filter_channels, kernel_size, padding=kernel_size//2)
112
  self.norm_2 = modules.LayerNorm(filter_channels)
113
  self.proj = nn.Conv1d(filter_channels, 1, 1)