We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4e2be2f commit 60f12e3Copy full SHA for 60f12e3
pytorch_lightning/utilities/auto_restart.py
@@ -344,7 +344,7 @@ def _dataloader_to_state_dict(
344
if not isinstance(dataloader.dataset, CaptureIterableDataset):
345
fast_forward_sampler = _find_fast_forward_samplers(dataloader)
346
if fast_forward_sampler is not None:
347
- samplers_state_dict = fast_forward_sampler.state_dict(num_batches_processed=num_batches_processed).items()
+ samplers_state_dict = fast_forward_sampler.state_dict(num_batches_processed=num_batches_processed)
348
for sampler_k, sampler_v in samplers_state_dict.items():
349
out[sampler_k] = sampler_v
350
return out
0 commit comments