This commit is contained in:
Starbeamrainbowlabs 2022-10-21 14:26:58 +01:00
parent 0e814b7e98
commit 847cd97ec4
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -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")