From ef0f15960d525f9e583097e69fab5ebf2a43ac1c Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 2 Sep 2022 18:09:36 +0100 Subject: [PATCH] send correct water shape to --- aimodel/src/subcommands/pretrain.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/aimodel/src/subcommands/pretrain.py b/aimodel/src/subcommands/pretrain.py index d1e2737..2d35fe6 100644 --- a/aimodel/src/subcommands/pretrain.py +++ b/aimodel/src/subcommands/pretrain.py @@ -1,3 +1,4 @@ +import math import sys import argparse from asyncio.log import logger @@ -52,7 +53,7 @@ def run(args): feature_dim=args.feature_dim, shape_rainfall=dataset_metadata["rainfallradar"], - shape_water=dataset_metadata["waterdepth"] + shape_water=[ math.floor(value * 0.75) for value in dataset_metadata["waterdepth"] ] ) ai.train(dataset_train, dataset_validate)