From cf872ef739be4fcd7b7074bd40351b4d8f9bdfc8 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Mon, 31 Oct 2022 18:40:58 +0000 Subject: [PATCH] how could I be so *stupid*...... --- aimodel/src/lib/ai/RainfallWaterContraster.py | 2 +- aimodel/src/subcommands/pretrain.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/aimodel/src/lib/ai/RainfallWaterContraster.py b/aimodel/src/lib/ai/RainfallWaterContraster.py index cce6469..154c5bf 100644 --- a/aimodel/src/lib/ai/RainfallWaterContraster.py +++ b/aimodel/src/lib/ai/RainfallWaterContraster.py @@ -85,7 +85,7 @@ class RainfallWaterContraster(object): validation_data=dataset_validate, epochs=self.epochs, 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): diff --git a/aimodel/src/subcommands/pretrain.py b/aimodel/src/subcommands/pretrain.py index 5a2f68a..753f547 100644 --- a/aimodel/src/subcommands/pretrain.py +++ b/aimodel/src/subcommands/pretrain.py @@ -48,8 +48,8 @@ def run(args): batch_size=args.batch_size, ) - print("BATCHES_TRAIN", count_batches(dataset_train)) - print("BATCHES_VALIDATE", count_batches(dataset_validate)) + # print("BATCHES_TRAIN", count_batches(dataset_train)) # 18500 for the full 2006-2020 dataset + # print("BATCHES_VALIDATE", count_batches(dataset_validate)) # 4653 for the full 2006-2020 dataset