Glenn Jocher
GitHub
3 years ago
1 changed files with 1 additions and 1 deletions
+ 1
- 1
train.py
View File
| 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, |
Loading…