Skip to content

Commit e6ed8fd

Browse files
authored
skip conv-bn fuse on validation
1 parent 3c96da9 commit e6ed8fd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

val.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,7 @@ def run(
137137
(save_dir / 'labels' if save_txt else save_dir).mkdir(parents=True, exist_ok=True) # make dir
138138

139139
# Load model
140-
model = DetectMultiBackend(weights, device=device, dnn=dnn, data=data, fp16=half)
140+
model = DetectMultiBackend(weights, device=device, dnn=dnn, data=data, fp16=half, fuse=False)
141141
stride, pt, jit, engine = model.stride, model.pt, model.jit, model.engine
142142
imgsz = check_img_size(imgsz, s=stride) # check image size
143143
half = model.fp16 # FP16 supported on limited backends with CUDA

0 commit comments

Comments
 (0)