mirrors
/
deit
mirror of https://github.com/facebookresearch/deit
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update resmlp_models.py

pull/118/head
Hugo Touvron 2021-06-18 20:59:28 +02:00 committed by GitHub
parent 6fa7ef60b4
commit 31b3d676b3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
resmlp_models.py
View File

@ -173,7 +173,7 @@ def resmlp_36(pretrained=False,dist=False, **kwargs):
return model return model
@register_model @register_model
def resmlpB_24(pretrained=False,dist=False, 22k = False, **kwargs): def resmlpB_24(pretrained=False,dist=False, in_22k = False, **kwargs):
model = resmlp_models( model = resmlp_models(
patch_size=8, embed_dim=768, depth=24, patch_size=8, embed_dim=768, depth=24,
Patch_layer=PatchEmbed, Patch_layer=PatchEmbed,
@ -182,7 +182,7 @@ def resmlpB_24(pretrained=False,dist=False, 22k = False, **kwargs):
if pretrained: if pretrained:
if dist: if dist:
url_path = "https://dl.fbaipublicfiles.com/deit/resmlpB_24_dist.pth" url_path = "https://dl.fbaipublicfiles.com/deit/resmlpB_24_dist.pth"
elif 22k: elif in_22k:
url_path = "https://dl.fbaipublicfiles.com/deit/resmlpB_24_22k.pth" url_path = "https://dl.fbaipublicfiles.com/deit/resmlpB_24_22k.pth"
else: else:
url_path = "https://dl.fbaipublicfiles.com/deit/resmlpB_24_no_dist.pth" url_path = "https://dl.fbaipublicfiles.com/deit/resmlpB_24_no_dist.pth"