未验证 提交 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 ...@@ -12,7 +12,7 @@ weights: output/fcos_r50_fpn_iou_1x_coco/model_final
TrainReader: TrainReader:
sample_transforms: sample_transforms:
- Decode: {} - 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} - NormalizeImage: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: True}
- RandomFlip: {} - RandomFlip: {}
batch_transforms: batch_transforms:
...@@ -26,6 +26,7 @@ TrainReader: ...@@ -26,6 +26,7 @@ TrainReader:
batch_size: 2 batch_size: 2
shuffle: True shuffle: True
drop_last: True drop_last: True
use_shared_memory: True
EvalReader: EvalReader:
......
...@@ -25,6 +25,7 @@ TrainReader: ...@@ -25,6 +25,7 @@ TrainReader:
batch_size: 2 batch_size: 2
shuffle: True shuffle: True
drop_last: True drop_last: True
use_shared_memory: True
EvalReader: EvalReader:
......
...@@ -25,6 +25,7 @@ TrainReader: ...@@ -25,6 +25,7 @@ TrainReader:
batch_size: 2 batch_size: 2
shuffle: True shuffle: True
drop_last: True drop_last: True
use_shared_memory: True
epoch: 24 epoch: 24
......
...@@ -43,6 +43,8 @@ DistillPPYOLOELoss: # L -> M ...@@ -43,6 +43,8 @@ DistillPPYOLOELoss: # L -> M
loss_weight: {'logits': 4.0, 'feat': 1.0} loss_weight: {'logits': 4.0, 'feat': 1.0}
logits_distill: True logits_distill: True
logits_loss_weight: {'class': 1.0, 'iou': 2.5, 'dfl': 0.5} 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_distill: True
feat_distiller: 'fgd' # ['cwd', 'fgd', 'pkd', 'mgd', 'mimic'] feat_distiller: 'fgd' # ['cwd', 'fgd', 'pkd', 'mgd', 'mimic']
feat_distill_place: 'neck_feats' feat_distill_place: 'neck_feats'
......
...@@ -43,6 +43,8 @@ DistillPPYOLOELoss: # X -> L ...@@ -43,6 +43,8 @@ DistillPPYOLOELoss: # X -> L
loss_weight: {'logits': 4.0, 'feat': 1.0} loss_weight: {'logits': 4.0, 'feat': 1.0}
logits_distill: True logits_distill: True
logits_loss_weight: {'class': 1.0, 'iou': 2.5, 'dfl': 0.5} 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_distill: True
feat_distiller: 'fgd' # ['cwd', 'fgd', 'pkd', 'mgd', 'mimic'] feat_distiller: 'fgd' # ['cwd', 'fgd', 'pkd', 'mgd', 'mimic']
feat_distill_place: 'neck_feats' feat_distill_place: 'neck_feats'
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册