Skip to content

GitLab

"src/array/cuda/gather_mm.hip" did not exist on "b3d3a2c4b0e04320bd9c0cedeb7ed70813d2d824"
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
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
torchaudio/models/wav2vec2/components.py
View file @ 2574e114
...@@ -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