This commit is contained in:
Starbeamrainbowlabs 2022-10-11 19:19:23 +01:00
parent 11f91a7cf4
commit f4a2c742d9
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -65,7 +65,7 @@ def run(args):
batch_size=args.batch_size,
feature_dim_in=args.feature_dim,
model_arch=args.model_arch,
model_arch=args.arch,
metadata = read_metadata(args.input),
shape_water_out=[ args.water_size, args.water_size ], # The DESIRED output shape. the actual data will be cropped to match this.
)