浏览代码

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,

正在加载...
取消
保存