facebookresearch/fairseq
1
1
Fork 0
mirror of https://github.com/facebookresearch/fairseq.git synced 2024-09-21 06:13:31 +03:00
Code Issues Packages Projects Releases Wiki Activity

make apex optional (#4906)

Browse Source
This commit is contained in:
Alexei Baevski 2022-12-13 00:50:53 -08:00 committed by GitHub
parent 902f4aa5e8
commit c19aed8ef5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
examples/data2vec/models/mae.py
View File

@ -21,7 +21,11 @@ from fairseq.dataclass import FairseqDataclass
from fairseq.models import BaseFairseqModel, register_model
from fairseq.models.wav2vec.wav2vec2 import TransformerSentenceEncoderLayer
from apex.normalization import FusedLayerNorm
try:
from apex.normalization import FusedLayerNorm
except:
FusedLayerNorm = nn.LayerNorm
import torch.nn.functional as F