sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-12-22 22:25: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 sys
import argparse
from asyncio.log import logger
import tensorflow as tf
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
def parse_args():
@ -51,7 +50,7 @@ def run(args):
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,
batch_size=args.batch_size,
water_threshold=args.water_threshold,