瀏覽代碼

Update train.py (#5014)

modifyDataloader
Glenn Jocher GitHub 3 年之前
父節點
當前提交
452e84703c
沒有發現已知的金鑰在資料庫的簽署中 GPG 金鑰 ID: 4AEE18F83AFDEB23
共有 1 個檔案被更改,包括 1 行新增1 行删除
  1. +1
    -1
      train.py

+ 1
- 1
train.py 查看文件

@@ -412,7 +412,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary
batch_size=batch_size // WORLD_SIZE * 2,
imgsz=imgsz,
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,
dataloader=val_loader,
save_dir=save_dir,

Loading…
取消
儲存