未验证 提交 a9cd917e 编写于 作者: F Feng Ni 提交者: GitHub

[cherry-pick] fix some configs (#7770)

上级 53d7e33c
......@@ -12,7 +12,7 @@ weights: output/fcos_r50_fpn_iou_1x_coco/model_final
TrainReader:
sample_transforms:
- Decode: {}
- RandomResize: {target_size: [[640, 1333], [672, 1333], [704, 1333], [736, 1333], [768, 1333], [800, 1333]], keep_ratio: True, interp: 1}
- Resize: {target_size: [800, 1333], keep_ratio: True, interp: 1}
- NormalizeImage: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: True}
- RandomFlip: {}
batch_transforms:
......@@ -26,6 +26,7 @@ TrainReader:
batch_size: 2
shuffle: True
drop_last: True
use_shared_memory: True
EvalReader:
......
......@@ -25,6 +25,7 @@ TrainReader:
batch_size: 2
shuffle: True
drop_last: True
use_shared_memory: True
EvalReader:
......
......@@ -25,6 +25,7 @@ TrainReader:
batch_size: 2
shuffle: True
drop_last: True
use_shared_memory: True
epoch: 24
......
......@@ -43,6 +43,8 @@ DistillPPYOLOELoss: # L -> M
loss_weight: {'logits': 4.0, 'feat': 1.0}
logits_distill: True
logits_loss_weight: {'class': 1.0, 'iou': 2.5, 'dfl': 0.5}
logits_ld_distill: True
logits_ld_params: {'weight': 20000, 'T': 10}
feat_distill: True
feat_distiller: 'fgd' # ['cwd', 'fgd', 'pkd', 'mgd', 'mimic']
feat_distill_place: 'neck_feats'
......
......@@ -43,6 +43,8 @@ DistillPPYOLOELoss: # X -> L
loss_weight: {'logits': 4.0, 'feat': 1.0}
logits_distill: True
logits_loss_weight: {'class': 1.0, 'iou': 2.5, 'dfl': 0.5}
logits_ld_distill: True
logits_ld_params: {'weight': 20000, 'T': 10}
feat_distill: True
feat_distiller: 'fgd' # ['cwd', 'fgd', 'pkd', 'mgd', 'mimic']
feat_distill_place: 'neck_feats'
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册