Update train.py (#5014)
This commit is contained in:
1 parent
cce313d67d
commit
452e84703c
1 file changed
+1
-1
@@ -412,7 +412,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary
|
|||||||
batch_size=batch_size // WORLD_SIZE * 2,
|
batch_size=batch_size // WORLD_SIZE * 2,
|
||||||
imgsz=imgsz,
|
imgsz=imgsz,
|
||||||
model=attempt_load(f, device).half(),
|
model=attempt_load(f, device).half(),
|
||||||
iou_thres=0.7 if is_coco else 0.6, # best pycocotools results at 0.7
|
iou_thres=0.65 if is_coco else 0.60, # best pycocotools results at 0.65
|
||||||
single_cls=single_cls,
|
single_cls=single_cls,
|
||||||
dataloader=val_loader,
|
dataloader=val_loader,
|
||||||
save_dir=save_dir,
|
save_dir=save_dir,
|
||||||
|
|||||||
Reference in new issue
Block a user