From f655f974eb6d2d67d984289497c072e070d98702 Mon Sep 17 00:00:00 2001 From: Jirka Borovec Date: Wed, 17 Feb 2021 06:52:07 +0100 Subject: [PATCH] hotfix: move process_dataloader to plugins (#6023) --- pytorch_lightning/trainer/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py index 2e45f9502e..2453a08ba9 100644 --- a/pytorch_lightning/trainer/trainer.py +++ b/pytorch_lightning/trainer/trainer.py @@ -823,7 +823,7 @@ class Trainer( # run validation/testing for dataloader_idx, dataloader in enumerate(dataloaders): - dataloader = self.accelerator_backend.process_dataloader(dataloader) + dataloader = self.training_type_plugin.process_dataloader(dataloader) dl_max_batches = self.predict_loop.max_batches[dataloader_idx] for batch_idx, batch in enumerate(dataloader):