Python _teardown_dataloader_get_iterators Exemples

Exemple #1

 def teardown(self) -> None:
     self.reset()
     if isinstance(self.dataloader, CombinedLoader):
         self.dataloader.reset()
     if isinstance(self.dataloader, DataLoader):
         CombinedLoader._shutdown_workers_and_reset_iterator(
             self.dataloader)
     self.dataloader_iter = None
     _teardown_dataloader_get_iterators()

Exemple #2

Fichier : data_connector.py Projet : oddeirikigland/pytorch-lightning

 def teardown(self) -> None:
     if self.train_data_fetcher:
         self.train_data_fetcher.teardown()
         self.train_data_fetcher = None
     if self.validate_data_fetcher:
         self.validate_data_fetcher.teardown()
         self.validate_data_fetcher = None
     if self.test_data_fetcher:
         self.test_data_fetcher.teardown()
         self.test_data_fetcher = None
     if self.sanity_check_data_fetcher:
         self.sanity_check_data_fetcher.teardown()
         self.sanity_check_data_fetcher = None
     _teardown_dataloader_get_iterators()