find_unused_parameters

游雁

2024-09-30 0bf8edca37d384eac6d0c345edd6b1a2b88f97ae

 funasr/models/transformer/utils/multi_layer_conv.py

@@ -96,7 +96,7 @@
            kernel_size,
            stride=1,
            padding=(kernel_size - 1) // 2,
            bias=False
            bias=False,
        )
        self.norm = torch.nn.LayerNorm(hidden_chans)
        self.dropout = torch.nn.Dropout(dropout_rate)