wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add tags to the rendezvous calls for TPU. (#921)

Browse Source 
* Update data_loading.py

* Update training_io.py

* Update trainer.py
This commit is contained in:
srush
2020-02-23 15:00:32 -05:00
committed by GitHub
parent 89d5772f55
commit 5778a4131c
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/trainer/data_loading.py
+1 -1
View File
@@ -233,7 +233,7 @@ class TrainerDataLoadingMixin(ABC):
self.get_val_dataloaders()
# wait for all processes to catch up
torch_xla.core.xla_model.rendezvous()
torch_xla.core.xla_model.rendezvous("pl.TrainerDataLoadingMixin.get_dataloaders")
# support IterableDataset for train data
self.is_iterable_train_dataloader = (