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

train_mono: fix crash

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-11-11 18:26:25 +00:00
parent 8ac5159adc
commit 65e801cf28
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 2 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
aimodel/src/subcommands/train_mono.py
View file

@ -1,12 +1,11 @@
import math import math
import sys import sys
import argparse import argparse
from asyncio.log import logger
import tensorflow as tf import tensorflow as tf
from lib.ai.RainfallWaterMono import RainfallWaterMono from lib.ai.RainfallWaterMono import RainfallWaterMono
from lib.dataset.dataset import dataset from lib.dataset.dataset_mono import dataset_mono
from lib.dataset.read_metadata import read_metadata from lib.dataset.read_metadata import read_metadata
def parse_args(): def parse_args():
@ -51,7 +50,7 @@ def run(args):
sys.stderr.write(f"\n\n>>> This is TensorFlow {tf.__version__}\n\n\n") sys.stderr.write(f"\n\n>>> This is TensorFlow {tf.__version__}\n\n\n")
dataset_train, dataset_validate = dataset( dataset_train, dataset_validate = dataset_mono(
dirpath_input=args.input, dirpath_input=args.input,
batch_size=args.batch_size, batch_size=args.batch_size,
water_threshold=args.water_threshold, water_threshold=args.water_threshold,