pytorch_lightning.trainer.data_loading module¶
-
class
pytorch_lightning.trainer.data_loading.
TrainerDataLoadingMixin
[source]¶ Bases:
abc.ABC
-
_reset_eval_dataloader
(model, mode)[source]¶ Generic method to reset a dataloader for evaluation.
- Parameters
model¶ (
LightningModule
) – The current LightningModule
- Return type
Tuple
[int
,List
[DataLoader
]]- Returns
Tuple (num_batches, dataloaders)
-
determine_data_use_amount
(train_percent_check, val_percent_check, test_percent_check, overfit_pct)[source]¶ Use less data for debugging purposes
- Return type
None
-
abstract
is_overridden
(*args)[source]¶ Warning: this is just empty shell for code implemented in other class.
-
request_dataloader
(dataloader_fx)[source]¶ Handles downloading data in the GPU or TPU case.
- Parameters
- Return type
- Returns
The dataloader
-
reset_test_dataloader
(model)[source]¶ Resets the validation dataloader and determines the number of batches.
- Parameters
model¶ – The current LightningModule
- Return type
None
-
reset_train_dataloader
(model)[source]¶ Resets the train dataloader and initialises required variables (number of batches, when to validate, etc.).
- Parameters
model¶ (
LightningModule
) – The current LightningModule- Return type
None
-
reset_val_dataloader
(model)[source]¶ Resets the validation dataloader and determines the number of batches.
- Parameters
model¶ (
LightningModule
) – The current LightningModule- Return type
None
-