Update trainer.py - fix DistributedSampler import (#93)

DistributedSampler is used but not imported in `trainer.py`
neuralmagic · Sep 25, 2023 · e25912d · e25912d
1 parent 6167de3
commit e25912d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/transformers/trainer.py b/src/transformers/trainer.py
@@ -51,6 +51,7 @@
 from packaging import version
 from torch import nn
 from torch.utils.data import DataLoader, Dataset, RandomSampler, SequentialSampler
+from torch.utils.data.distributed import DistributedSampler
 
 from . import __version__
 from .configuration_utils import PretrainedConfig