diff --git a/configs/fcos/fcos_r50_fpn_iou_1x_coco.yml b/configs/fcos/fcos_r50_fpn_iou_1x_coco.yml index 943c5bc04dedb4759ef64b8eeffd2e1b1b074fb2..18c33cf8e221af72ca3edb1ad355572ee456a3ae 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 3f6a327db268501c05447a02805753012b76e9b3..d53ea17f57b6c78076668489ca5246122bfe8edb 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 85a9781359deaaa3c1fe87e8d46376bd72c348c9..0afdbbc5be62468ae073258badb0ee2773948e3c 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 dbef4902e90594390862078ae32957a8c83b19e6..0a5bfcd29cc352bb67f48a33a28f4531728543e5 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 3f78deaefe7e6059ae493fe3117842414ef5f37e..55d3c4c9f08ae373e764062c3820f3823710e1d7 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'