File tree Expand file tree Collapse file tree 1 file changed +6
-6
lines changed Expand file tree Collapse file tree 1 file changed +6
-6
lines changed Original file line number Diff line number Diff 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 (
You can’t perform that action at this time.
0 commit comments