Skip to content

Commit 90c5bc9

Browse files
authored
Resolve the issue of parameter accuracy in dev branches (#3108)
1 parent 2dd529a commit 90c5bc9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddleformers/trainer/trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1833,7 +1833,7 @@ def _inner_training_loop(
18331833
if (
18341834
not args.enable_auto_parallel
18351835
and isinstance(train_dataloader, paddle.io.DataLoader)
1836-
and isinstance(train_dataloader.batch_sampler, (DistributedBatchSampler, NlpDistributedBatchSampler))
1836+
and isinstance(train_dataloader.batch_sampler, DistributedBatchSampler)
18371837
):
18381838
train_dataloader.batch_sampler.set_epoch(epoch)
18391839

0 commit comments

Comments
 (0)