From a9cd917e68ea9ee3ebda8dc47d73de4eadf55f19 Mon Sep 17 00:00:00 2001 From: Feng Ni Date: Wed, 15 Feb 2023 18:31:17 +0800 Subject: [PATCH] [cherry-pick] fix some configs (#7770) --- configs/fcos/fcos_r50_fpn_iou_1x_coco.yml | 3 ++- configs/fcos/fcos_r50_fpn_iou_multiscale_2x_coco.yml | 1 + configs/fcos/fcos_r50_fpn_multiscale_2x_coco.yml | 1 + configs/slim/distill/ppyoloe_plus_distill_l_distill_m.yml | 2 ++ configs/slim/distill/ppyoloe_plus_distill_x_distill_l.yml | 2 ++ 5 files changed, 8 insertions(+), 1 deletion(-) diff --git a/configs/fcos/fcos_r50_fpn_iou_1x_coco.yml b/configs/fcos/fcos_r50_fpn_iou_1x_coco.yml index 943c5bc04..18c33cf8e 100644 --- a/configs/fcos/fcos_r50_fpn_iou_1x_coco.yml +++ b/configs/fcos/fcos_r50_fpn_iou_1x_coco.yml @@ -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: diff --git a/configs/fcos/fcos_r50_fpn_iou_multiscale_2x_coco.yml b/configs/fcos/fcos_r50_fpn_iou_multiscale_2x_coco.yml index 3f6a327db..d53ea17f5 100644 --- a/configs/fcos/fcos_r50_fpn_iou_multiscale_2x_coco.yml +++ b/configs/fcos/fcos_r50_fpn_iou_multiscale_2x_coco.yml @@ -25,6 +25,7 @@ TrainReader: batch_size: 2 shuffle: True drop_last: True + use_shared_memory: True EvalReader: diff --git a/configs/fcos/fcos_r50_fpn_multiscale_2x_coco.yml b/configs/fcos/fcos_r50_fpn_multiscale_2x_coco.yml index 85a978135..0afdbbc5b 100644 --- a/configs/fcos/fcos_r50_fpn_multiscale_2x_coco.yml +++ b/configs/fcos/fcos_r50_fpn_multiscale_2x_coco.yml @@ -25,6 +25,7 @@ TrainReader: batch_size: 2 shuffle: True drop_last: True + use_shared_memory: True epoch: 24 diff --git a/configs/slim/distill/ppyoloe_plus_distill_l_distill_m.yml b/configs/slim/distill/ppyoloe_plus_distill_l_distill_m.yml index dbef4902e..0a5bfcd29 100644 --- a/configs/slim/distill/ppyoloe_plus_distill_l_distill_m.yml +++ b/configs/slim/distill/ppyoloe_plus_distill_l_distill_m.yml @@ -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' diff --git a/configs/slim/distill/ppyoloe_plus_distill_x_distill_l.yml b/configs/slim/distill/ppyoloe_plus_distill_x_distill_l.yml index 3f78deaef..55d3c4c9f 100644 --- a/configs/slim/distill/ppyoloe_plus_distill_x_distill_l.yml +++ b/configs/slim/distill/ppyoloe_plus_distill_x_distill_l.yml @@ -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' -- GitLab