未验证 提交 339fe716 编写于 作者: W Wenyu 提交者: GitHub

update drop_path_rate (#4101)

上级 aff7391f
...@@ -13,10 +13,9 @@ SwinTransformer: ...@@ -13,10 +13,9 @@ SwinTransformer:
num_heads: [3, 6, 12, 24] num_heads: [3, 6, 12, 24]
window_size: 7 window_size: 7
ape: false ape: false
drop_path_rate: 0.1 drop_path_rate: 0.2
patch_norm: true patch_norm: true
out_indices: [0,1,2,3] 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 pretrained: https://paddledet.bj.bcebos.com/models/pretrained/swin_tiny_patch4_window7_224.pdparams
FPN: FPN:
......
...@@ -673,9 +673,6 @@ class SwinTransformer(nn.Layer): ...@@ -673,9 +673,6 @@ class SwinTransformer(nn.Layer):
else: #model in local path else: #model in local path
path = pretrained path = pretrained
self.set_state_dict(paddle.load(path)) self.set_state_dict(paddle.load(path))
print('###################################################')
print('###############Success load the mode###############')
print('###################################################')
def _freeze_stages(self): def _freeze_stages(self):
if self.frozen_stages >= 0: if self.frozen_stages >= 0:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册