sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-11-22 09:13:01 +00:00
Code Issues Projects Releases Packages Wiki Activity

how could I be so *stupid*......

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-10-31 18:40:58 +00:00
parent da32d75778
commit cf872ef739
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
aimodel/src/lib/ai/RainfallWaterContraster.py
View file

@ -85,7 +85,7 @@ class RainfallWaterContraster(object):
validation_data=dataset_validate, validation_data=dataset_validate,
epochs=self.epochs, epochs=self.epochs,
callbacks=make_callbacks(self.dir_output, self.model_predict), callbacks=make_callbacks(self.dir_output, self.model_predict),
steps_per_epoch=10 # For testing # steps_per_epoch=10 # For testing
) )
def embed(self, dataset): def embed(self, dataset):

4
aimodel/src/subcommands/pretrain.py
View file

@ -48,8 +48,8 @@ def run(args):
batch_size=args.batch_size, batch_size=args.batch_size,
) )
print("BATCHES_TRAIN", count_batches(dataset_train)) # print("BATCHES_TRAIN", count_batches(dataset_train)) # 18500 for the full 2006-2020 dataset
print("BATCHES_VALIDATE", count_batches(dataset_validate)) # print("BATCHES_VALIDATE", count_batches(dataset_validate)) # 4653 for the full 2006-2020 dataset