yolov5_m_300e_coco.yml 406 Bytes
Newer Older
dlyrm's avatar
dlyrm committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
_BASE_: [
  '../datasets/coco_detection.yml',
  '../runtime.yml',
  '_base_/optimizer_300e_high.yml',
  '_base_/yolov5_cspdarknet.yml',
  '_base_/yolov5_reader_high_aug.yml',
]
depth_mult: 0.67
width_mult: 0.75

log_iter: 100
snapshot_epoch: 10
weights: output/yolov5_m_300e_coco/model_final


TrainReader:
  batch_size: 16 # default 8 gpus, total bs = 128


YOLOv5Loss:
  obj_weight: 0.7
  cls_weght: 0.3