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

train_mono_predict: limit label size to 64x64

Browse source 
that's the size the model predicts
This commit is contained in:
Starbeamrainbowlabs 2022-11-25 17:47:17 +00:00
parent 51dd484d13
commit e7410fb480
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
aimodel/src/subcommands/train_mono_predict.py
View file

@ -74,7 +74,8 @@ def run(args):
# Note that if using a directory of input files, the output order is NOT GUARANTEED TO BE THE SAME. In fact, it probably won't be (see dataset_mono for more details).
dataset = dataset_mono_predict(
dirpath_input=args.input,
parallel_reads_multiplier=args.read_multiplier
parallel_reads_multiplier=args.read_multiplier,
shape_water_desired=[64, 64]
)
# for items in dataset_train.repeat(10):