Commit 2574e114 authored by Zhaoheng Ni's avatar Zhaoheng Ni
Browse files

Fix initialization in hubert_pretrain_model (#2846)

Summary:
address https://github.com/pytorch/audio/issues/2845

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

Reviewed By: carolineechen

Differential Revision: D41251624

Pulled By: nateanl

fbshipit-source-id: 1a363d2314d6a452f35c109b9730da64ada5a2fd
parent 71ddee16
...@@ -197,6 +197,7 @@ class ConvolutionalPositionalEmbedding(Module): ...@@ -197,6 +197,7 @@ class ConvolutionalPositionalEmbedding(Module):
): ):
super().__init__() super().__init__()
self.embed_dim = embed_dim self.embed_dim = embed_dim
self.kernel_size = kernel_size
self.conv = nn.Conv1d( self.conv = nn.Conv1d(
in_channels=embed_dim, in_channels=embed_dim,
out_channels=embed_dim, out_channels=embed_dim,
......
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