mirror of
https://github.com/sbrl/research-rainfallradar
synced 2024-11-22 01:12:59 +00:00
dlr ds/mono: just why
This commit is contained in:
parent
a4e80229fb
commit
9efc72db73
1 changed files with 1 additions and 1 deletions
|
@ -175,7 +175,7 @@ def dataset_mono_predict(dirpath_input, batch_size=64, **kwargs):
|
|||
return make_dataset(
|
||||
filepaths=filepaths,
|
||||
metadata=read_metadata(dirpath_input),
|
||||
batch_size=None, # NOTE: We IGNORE batch_size because otherwise we have issues reading the dataset which is stupid because it doesn't always happen and it's weird.
|
||||
batch_size=batch_size, # WAS None
|
||||
shuffle=False, #even with shuffle=False we're not gonna get them all in the same order since we're reading in parallel
|
||||
**kwargs
|
||||
)
|
||||
|
|
Loading…
Reference in a new issue