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

update drop_path_rate (#4101)

上级 aff7391f
......@@ -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:
......
......@@ -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:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册