"docs/vscode:/vscode.git/clone" did not exist on "8125372bbdf2c35a6ac7361b6343d9a3d175ca1c"
Commit 738d2f8e authored by hwangjeff's avatar hwangjeff Committed by Facebook GitHub Bot
Browse files

Pass bias and dropout args to Conformer convolution block (#2215)

Summary:
Modifies `ConformerLayer` to pass `bias=True` (to be consistent with feedforward network defaults) and `dropout=dropout` (omission was a bug) to the convolution block.

Pull Request resolved: https://github.com/pytorch/audio/pull/2215

Reviewed By: carolineechen, nateanl

Differential Revision: D34164345

Pulled By: hwangjeff

fbshipit-source-id: 59fc804a1fe3b96e69e9fa5a2f9de94194d7bc55
parent 16d02a9e
......@@ -144,6 +144,8 @@ class ConformerLayer(torch.nn.Module):
input_dim=input_dim,
num_channels=input_dim,
depthwise_kernel_size=depthwise_conv_kernel_size,
bias=True,
dropout=dropout,
)
self.ffn2 = _FeedForwardModule(input_dim, ffn_dim, dropout=dropout)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment