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

Precise description of reload_dataloaders_every_n_epochs (#14245)

This commit is contained in:
Benjamin Krala 2022-09-01 16:06:28 +02:00 committed by GitHub
parent 28e18881a9
commit cce55b6cd3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
docs/source-pytorch/common/trainer.rst
View File

@ -1232,12 +1232,15 @@ reload_dataloaders_every_n_epochs
|
Set to a positive integer to reload dataloaders every n epochs.
Set to a positive integer to reload dataloaders every n epochs from your currently used data source.
DataSource can be a ``LightningModule`` or a ``LightningDataModule``.
.. code-block:: python
# if 0 (default)
train_loader = model.train_dataloader()
# or if using data module: datamodule.train_dataloader()
for epoch in epochs:
for batch in train_loader:
...
@ -1246,9 +1249,12 @@ Set to a positive integer to reload dataloaders every n epochs.
for epoch in epochs:
if not epoch % reload_dataloaders_every_n_epochs:
train_loader = model.train_dataloader()
# or if using data module: datamodule.train_dataloader()
for batch in train_loader:
...
The pseudocode applies also to the ``val_dataloader``.
.. _replace-sampler-ddp:
replace_sampler_ddp