From 847cd97ec47e4a34ec773bc1a3cd42dac0426514 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 21 Oct 2022 14:26:58 +0100 Subject: [PATCH] fixup --- aimodel/src/subcommands/train_predict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/subcommands/train_predict.py b/aimodel/src/subcommands/train_predict.py index 5bd0e95..c47637b 100644 --- a/aimodel/src/subcommands/train_predict.py +++ b/aimodel/src/subcommands/train_predict.py @@ -59,7 +59,7 @@ def run(args): os.mkdir(dirpath_output) - ai = RainfallWaterContraster.from_checkpoint(args.checkpoint, **json.loads(readfile(args.params))) + ai = RainfallWaterSegmenter.from_checkpoint(args.checkpoint, **json.loads(readfile(args.params))) sys.stderr.write(f"\n\n>>> This is TensorFlow {tf.__version__}\n\n\n")