sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-11-21 17:03:00 +00:00
Code Issues Projects Releases Packages Wiki Activity

dlr ds/mono: fixup

Browse source
This commit is contained in:
Starbeamrainbowlabs 2023-06-16 18:35:02 +01:00
parent 7869505cfb
commit a4e80229fb
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
aimodel/src/lib/dataset/dataset_mono.py
View file

@ -158,7 +158,7 @@ def dataset_mono(dirpath_input, train_percentage=0.8, **kwargs):
return dataset_train, dataset_validate #, filepaths
def dataset_mono_predict(dirpath_input, **kwargs):
def dataset_mono_predict(dirpath_input, batch_size=64, **kwargs):
"""Creates a tf.data.Dataset() for prediction using the contrastive learning model.
Note that this WILL MANGLE THE ORDERING if you set parallel_reads_multiplier to anything other than 0!!
@ -175,7 +175,7 @@ def dataset_mono_predict(dirpath_input, **kwargs):
return make_dataset(
filepaths=filepaths,
metadata=read_metadata(dirpath_input),
batch_size=None,
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.
shuffle=False, #even with shuffle=False we're not gonna get them all in the same order since we're reading in parallel
**kwargs
)