lightning/pytorch_lightning/loops
Yifu Wang 938a191406
Add a flavor of training_step that takes dataloader_iter as an argument (#8807)
* Add a flavor of training_step that takes dataloader_iter as an argument
2021-08-16 19:01:09 +00:00
..
batch Add a flavor of training_step that takes dataloader_iter as an argument (#8807) 2021-08-16 19:01:09 +00:00
dataloader Remove write_predictions from LightningModule (#8850) 2021-08-14 02:00:23 +00:00
epoch Add a flavor of training_step that takes dataloader_iter as an argument (#8807) 2021-08-16 19:01:09 +00:00
processors Add a flavor of training_step that takes dataloader_iter as an argument (#8807) 2021-08-16 19:01:09 +00:00
__init__.py Add a flavor of training_step that takes dataloader_iter as an argument (#8807) 2021-08-16 19:01:09 +00:00
base.py Save the `ResultCollection` in the loops state dict (#8641) 2021-08-02 20:52:24 +00:00
fit_loop.py Integrate `total_batch_idx` with progress tracking (#8598) 2021-08-14 14:08:34 +02:00
utilities.py Add a flavor of training_step that takes dataloader_iter as an argument (#8807) 2021-08-16 19:01:09 +00:00