This commit is contained in:
Yang Nie 2022-11-24 03:16:25 +08:00 committed by Tingquan Gao
parent a2052232e6
commit 1433161edd

View File

@ -688,5 +688,5 @@ def CvT_W24_384(pretrained=False, use_ssld=False, **kwargs):
spec=msvit_spec,
**kwargs)
_load_pretrained(
pretrained, model, MODEL_URLS["CvT_21_384"], use_ssld=use_ssld)
pretrained, model, MODEL_URLS["CvT_W24_384"], use_ssld=use_ssld)
return model