From 339fe7168a8982588c2fed10bd4cea4ea9947565 Mon Sep 17 00:00:00 2001 From: Wenyu Date: Sun, 5 Sep 2021 10:20:42 +0800 Subject: [PATCH] update drop_path_rate (#4101) --- configs/faster_rcnn/_base_/faster_rcnn_swin_transformer.yml | 3 +-- ppdet/modeling/backbones/swin_transformer.py | 3 --- 2 files changed, 1 insertion(+), 5 deletions(-) diff --git a/configs/faster_rcnn/_base_/faster_rcnn_swin_transformer.yml b/configs/faster_rcnn/_base_/faster_rcnn_swin_transformer.yml index 918fc52f1..5cae9ba7b 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 f4411b2f5..8ee8bd66a 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: -- GitLab