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 2dd529a commit 90c5bc9Copy full SHA for 90c5bc9
paddleformers/trainer/trainer.py
@@ -1833,7 +1833,7 @@ def _inner_training_loop(
1833
if (
1834
not args.enable_auto_parallel
1835
and isinstance(train_dataloader, paddle.io.DataLoader)
1836
- and isinstance(train_dataloader.batch_sampler, (DistributedBatchSampler, NlpDistributedBatchSampler))
+ and isinstance(train_dataloader.batch_sampler, DistributedBatchSampler)
1837
):
1838
train_dataloader.batch_sampler.set_epoch(epoch)
1839
0 commit comments