Преглед изворни кода

Merge pull request #796 from alibaba-damo-academy/dev_lhn

fix uniasr finetune bug
hnluo пре 2 година
родитељ
комит
0b359919f1
1 измењених фајлова са 2 додато и 1 уклоњено
  1. 2 1
      funasr/modules/subsampling.py

+ 2 - 1
funasr/modules/subsampling.py

@@ -358,7 +358,8 @@ class Conv1dSubsampling(torch.nn.Module):
         """
         """
         x = x.transpose(1, 2)  # (b, d ,t)
         x = x.transpose(1, 2)  # (b, d ,t)
         x = self.pad_fn(x)
         x = self.pad_fn(x)
-        x = F.relu(self.conv(x))
+        #x = F.relu(self.conv(x))
+        x = F.leaky_relu(self.conv(x), negative_slope=0.)
         x = x.transpose(1, 2)  # (b, t ,d)
         x = x.transpose(1, 2)  # (b, t ,d)
 
 
         if x_len is None:
         if x_len is None: