提交 4786b2c9 编写于 作者: weixin_46524038's avatar weixin_46524038 提交者: cuicheng01

add swinv2

上级 74de24f6
......@@ -893,7 +893,7 @@ def SwinTransformerV2_base_patch4_window16_256(
use_imagenet22k_pretrained=False,
use_imagenet22kto1k_pretrained=False,
**kwargs):
model = SwinTransformerV2(
model = SwinTransformerV2(
img_size=256,
embed_dim=128,
depths=[2, 2, 18, 2],
......@@ -984,4 +984,4 @@ def SwinTransformerV2_large_patch4_window24_384(
use_ssld=use_ssld,
use_imagenet22k_pretrained=use_imagenet22k_pretrained,
use_imagenet22kto1k_pretrained=use_imagenet22kto1k_pretrained)
return model
return model
\ No newline at end of file
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册