diff --git a/train.py b/train.py index 5bb5e1fe610b..e5758d334521 100644 --- a/train.py +++ b/train.py @@ -319,7 +319,7 @@ def lf(x): rect=True, rank=-1, workers=workers * 2, - pad=0.5, + pad=0.0, prefix=colorstr("val: "), )[0] diff --git a/val.py b/val.py index 600353b1c83d..cf88f9f09040 100644 --- a/val.py +++ b/val.py @@ -299,7 +299,7 @@ def run( f"classes). Pass correct combination of --weights and --data that are trained together." ) model.warmup(imgsz=(1 if pt else batch_size, 3, imgsz, imgsz)) # warmup - pad, rect = (0.0, False) if task == "speed" else (0.5, pt) # square inference for benchmarks + rect = False if task == "speed" else pt # square inference for benchmarks task = task if task in ("train", "val", "test") else "val" # path to train/val/test images dataloader = create_dataloader( data[task], @@ -307,7 +307,6 @@ def run( batch_size, stride, single_cls, - pad=pad, rect=rect, workers=workers, prefix=colorstr(f"{task}: "),