提交 4e87fbe7 编写于 作者: D dengkaipeng

update config

上级 825cea1d
...@@ -134,8 +134,9 @@ TrainReader: ...@@ -134,8 +134,9 @@ TrainReader:
downsample_ratios: [8, 16, 32] downsample_ratios: [8, 16, 32]
iou_thresh: 0.213 iou_thresh: 0.213
batch_size: 8 batch_size: 8
mosaic_prob: 0.3 mixup_epoch: 250
mosaic_epoch: 200 mosaic_prob: 0.5
mosaic_epoch: 260
shuffle: true shuffle: true
drop_last: true drop_last: true
worker_num: 8 worker_num: 8
......
architecture: YOLOv4 architecture: YOLOv4
use_gpu: true use_gpu: true
max_iters: 70000 max_iters: 72000
log_smooth_window: 20 log_smooth_window: 20
save_dir: output save_dir: output
snapshot_iter: 2000 snapshot_iter: 2000
...@@ -135,8 +135,8 @@ TrainReader: ...@@ -135,8 +135,8 @@ TrainReader:
iou_thresh: 0.213 iou_thresh: 0.213
batch_size: 8 batch_size: 8
mixup_epoch: 250 mixup_epoch: 250
mosaic_prob: 0.3 mosaic_prob: 0.5
mosaic_epoch: 300 mosaic_epoch: 260
shuffle: true shuffle: true
drop_last: true drop_last: true
worker_num: 8 worker_num: 8
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册