diff --git a/configs/faster_rcnn/_base_/faster_rcnn_swin_transformer.yml b/configs/faster_rcnn/_base_/faster_rcnn_swin_transformer.yml index 918fc52f1b3b9f4b92c5fd31c971bb4949f01cc1..5cae9ba7b628a0168807850150e0d752f261a6ab 100644 --- a/configs/faster_rcnn/_base_/faster_rcnn_swin_transformer.yml +++ b/configs/faster_rcnn/_base_/faster_rcnn_swin_transformer.yml @@ -13,10 +13,9 @@ SwinTransformer: num_heads: [3, 6, 12, 24] window_size: 7 ape: false - drop_path_rate: 0.1 + drop_path_rate: 0.2 patch_norm: true out_indices: [0,1,2,3] - drop_path_rate: 0.1 pretrained: https://paddledet.bj.bcebos.com/models/pretrained/swin_tiny_patch4_window7_224.pdparams FPN: diff --git a/ppdet/modeling/backbones/swin_transformer.py b/ppdet/modeling/backbones/swin_transformer.py index f4411b2f527f1e968c27be42b34f954f772d1e53..8ee8bd66a9e2aafa0bcc50ac19f1cad6a1cbc767 100644 --- a/ppdet/modeling/backbones/swin_transformer.py +++ b/ppdet/modeling/backbones/swin_transformer.py @@ -673,9 +673,6 @@ class SwinTransformer(nn.Layer): else: #model in local path path = pretrained self.set_state_dict(paddle.load(path)) - print('###################################################') - print('###############Success load the mode###############') - print('###################################################') def _freeze_stages(self): if self.frozen_stages >= 0: