mrq/torchscale
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix example fsdp

Browse Source
This commit is contained in:
shumingma 2022-12-08 04:20:27 -08:00
parent 6d62bbbf67
commit 2518ea030c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/fairseq/models/machine_translation.py
View File

@ -11,7 +11,7 @@ from typing import Dict, List, Optional, Tuple
import torch
from fairseq import distributed_utils, utils
from fairseq.distributed import utils as fsdp_wrap
from fairseq.distributed import fsdp_wrap
from fairseq.models import (
FairseqEncoder,
FairseqEncoderDecoderModel,