Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

hotfix: move process_dataloader to plugins (#6023)

This commit is contained in:
Jirka Borovec 2021-02-17 06:52:07 +01:00 committed by GitHub
parent ec0eac1197
commit f655f974eb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch_lightning/trainer/trainer.py
View File

@ -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):