Ver código fonte

Update train.py (#5014)

modifyDataloader
Glenn Jocher GitHub 3 anos atrás
pai
commit
452e84703c
Nenhuma chave conhecida encontrada para esta assinatura no banco de dados ID da chave GPG: 4AEE18F83AFDEB23
1 arquivos alterados com 1 adições e 1 exclusões
  1. +1
    -1
      train.py

+ 1
- 1
train.py Ver arquivo

@@ -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,

Carregando…
Cancelar
Salvar