fix

pytorch · Jul 31, 2023 · 4a5cdc0 · 4a5cdc0
1 parent 75fb860
commit 4a5cdc0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchaudio/models/wav2vec2/components.py b/torchaudio/models/wav2vec2/components.py
@@ -212,7 +212,7 @@ def __init__(
         self.num_remove: int = 1 if kernel_size % 2 == 0 else 0
 
     def __prepare_scriptable__(self):
-        if self.conv.__class__.__name__ == "ParameterizedConv1d":
+        if self.conv.__class__.__name__ == "ParametrizedConv1d":
             _LG.warning("Removing weight_norm from %s", self.__class__.__name__)
             torch.nn.utils.parametrize.remove_parametrizations(self.conv, "weight")
         return self