Skip to content

Commit 3c96da9

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

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -438,7 +438,7 @@ def train(hyp, opt, device, callbacks): # hyp is path/to/hyp.yaml or hyp dictio
438438
data_dict,
439439
batch_size=batch_size // WORLD_SIZE * 2,
440440
imgsz=imgsz,
441-
model=attempt_load(f, device).half(),
441+
model=attempt_load(f, device, fuse=False).half(),
442442
iou_thres=0.65 if is_coco else 0.60, # best pycocotools results at 0.65
443443
single_cls=single_cls,
444444
dataloader=val_loader,

0 commit comments

Comments
 (0)