Skip to content

Commit a9cd917

Browse files
authored
[cherry-pick] fix some configs (#7770)
1 parent 53d7e33 commit a9cd917

5 files changed

+8
-1
lines changed

configs/fcos/fcos_r50_fpn_iou_1x_coco.yml

+2-1
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ weights: output/fcos_r50_fpn_iou_1x_coco/model_final
1212
TrainReader:
1313
sample_transforms:
1414
- Decode: {}
15-
- RandomResize: {target_size: [[640, 1333], [672, 1333], [704, 1333], [736, 1333], [768, 1333], [800, 1333]], keep_ratio: True, interp: 1}
15+
- Resize: {target_size: [800, 1333], keep_ratio: True, interp: 1}
1616
- NormalizeImage: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: True}
1717
- RandomFlip: {}
1818
batch_transforms:
@@ -26,6 +26,7 @@ TrainReader:
2626
batch_size: 2
2727
shuffle: True
2828
drop_last: True
29+
use_shared_memory: True
2930

3031

3132
EvalReader:

configs/fcos/fcos_r50_fpn_iou_multiscale_2x_coco.yml

+1
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ TrainReader:
2525
batch_size: 2
2626
shuffle: True
2727
drop_last: True
28+
use_shared_memory: True
2829

2930

3031
EvalReader:

configs/fcos/fcos_r50_fpn_multiscale_2x_coco.yml

+1
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ TrainReader:
2525
batch_size: 2
2626
shuffle: True
2727
drop_last: True
28+
use_shared_memory: True
2829

2930
epoch: 24
3031

configs/slim/distill/ppyoloe_plus_distill_l_distill_m.yml

+2
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,8 @@ DistillPPYOLOELoss: # L -> M
4343
loss_weight: {'logits': 4.0, 'feat': 1.0}
4444
logits_distill: True
4545
logits_loss_weight: {'class': 1.0, 'iou': 2.5, 'dfl': 0.5}
46+
logits_ld_distill: True
47+
logits_ld_params: {'weight': 20000, 'T': 10}
4648
feat_distill: True
4749
feat_distiller: 'fgd' # ['cwd', 'fgd', 'pkd', 'mgd', 'mimic']
4850
feat_distill_place: 'neck_feats'

configs/slim/distill/ppyoloe_plus_distill_x_distill_l.yml

+2
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,8 @@ DistillPPYOLOELoss: # X -> L
4343
loss_weight: {'logits': 4.0, 'feat': 1.0}
4444
logits_distill: True
4545
logits_loss_weight: {'class': 1.0, 'iou': 2.5, 'dfl': 0.5}
46+
logits_ld_distill: True
47+
logits_ld_params: {'weight': 20000, 'T': 10}
4648
feat_distill: True
4749
feat_distiller: 'fgd' # ['cwd', 'fgd', 'pkd', 'mgd', 'mimic']
4850
feat_distill_place: 'neck_feats'

0 commit comments

Comments
 (0)