Commit b063723d authored by Gabriele Franch's avatar Gabriele Franch
Browse files

fixed batch size in padding

parent 60e07353
......@@ -128,6 +128,7 @@ def schedule_sampling(eta: float, itr: int):
)
return eta, real_input_flag
def wrapper_train(model: Model):
eta = args.sampling_start_value
iterator = get_batcher(args)
......
......@@ -42,7 +42,7 @@ def padding_taasss(array: np.ndarray, args) -> np.ndarray:
to
(1, 25, 480, 480, 1)
"""
zeros = np.zeros((1, 25, args.img_width, args.img_width, 1))
zeros = np.zeros((args.batch_size, 25, args.img_width, args.img_width, 1))
if args.img_width == 512:
zeros[:, :, 16:496, 16:496, :] = array
elif args.img_width == 256:
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment