Skip to content

Commit 6c6af64

Browse files
committed
Update base for Update on "Add Wav2Vec2 HuggingFace support"
This is not actually enabled in the benchmark suite as you need #103001 and also training is broken per #101160 but might as well review this part first. Contains #102979 but I will probably rebase past that once it lands. Signed-off-by: Edward Z. Yang <ezyangmeta.com> cc voznesenskym penguinwu anijain2305 EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx ipiszy bertmaher [ghstack-poisoned]
1 parent 9c9e8f3 commit 6c6af64

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

torch/nn/utils/weight_norm.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
from torch.nn.parameter import Parameter, UninitializedParameter
55
from torch import _weight_norm, norm_except_dim
66
from typing import Any, TypeVar
7+
import warnings
78
from ..modules import Module
89

910
__all__ = ['WeightNorm', 'weight_norm', 'remove_weight_norm']

0 commit comments

Comments
 (0)