Skip to content

Commit

Permalink
typo
Browse files Browse the repository at this point in the history
  • Loading branch information
FabianIsensee committed Mar 29, 2022
1 parent ea01899 commit b2dce1d
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions batchgenerators/dataloading/data_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -145,7 +145,7 @@ def __init__(self, data, batch_size, num_threads_in_multithreaded=1, seed_for_sh
:param seed_for_shuffle: for reproducibility
:param return_incomplete: whether or not to return batches that are incomplete. Only applies is infinite=False.
If your data has len of 34 and your batch size is 32 then there return_incomplete=False will make this loader
return only onebatch of shapre 32 (omitting 2 of your training examples). If return_incomplete=True a second
return only one batch of shape 32 (omitting 2 of your training examples). If return_incomplete=True a second
batch with batch size 2 will be returned.
:param shuffle: if True, the order of the indices will be shuffled between epochs. Only applies if infinite=False
:param infinite: if True, each batch contains randomly (uniformly) sampled indices. An unlimited number of
Expand All @@ -164,10 +164,10 @@ def __init__(self, data, batch_size, num_threads_in_multithreaded=1, seed_for_sh
self.current_position = None
self.was_initialized = False
self.last_reached = False
self.sampling_probabilities = sampling_probabilities

# when you derive, make sure to set this! We can't set it here because we don't know what data will be like
self.indices = None
self.sampling_probabilities = sampling_probabilities

def reset(self):
assert self.indices is not None
Expand Down

0 comments on commit b2dce1d

Please sign in to comment.