We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 03f9e01 commit 6b5f15bCopy full SHA for 6b5f15b
torchrec/distributed/train_pipeline/tracing.py
@@ -12,7 +12,6 @@
12
from typing import Any, Dict, List, Optional, Set, Tuple, Union
13
14
import torch
15
-
16
from torch.distributed._composable.fsdp.fully_shard import FSDPModule as FSDP2
17
18
from torch.distributed.fsdp import FullyShardedDataParallel as FSDP
0 commit comments