Skip to content

Commit 10d8c9a

Browse files
author
Eric Kerfoot
committed
Update epoch length block
Signed-off-by: Eric Kerfoot <eric.kerfoot@gmail>
1 parent dfcc77c commit 10d8c9a

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

monai/engines/workflow.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -133,12 +133,12 @@ def __init__(
133133
def set_sampler_epoch(engine: Engine) -> None:
134134
sampler.set_epoch(engine.state.epoch)
135135

136-
# if the epoch_length isn't given, attempt to get it from the length of the data loader
137-
if epoch_length is None:
138-
try:
139-
epoch_length = len(data_loader)
140-
except TypeError: # raised when data_loader is given an iterable dataset which has no length
141-
pass # deliberately leave epoch_length as None
136+
# if the epoch_length isn't given, attempt to get it from the length of the data loader
137+
if epoch_length is None:
138+
try:
139+
epoch_length = len(data_loader)
140+
except TypeError: # raised when data_loader has an iterable dataset with no length, or is some other type
141+
pass # deliberately leave epoch_length as None
142142

143143
# set all sharable data for the workflow based on Ignite engine.state
144144
self.state: Any = State(

0 commit comments

Comments
 (0)