update se_resnext_vd.py
parent
c719a3d74c
commit
5dad991c2a
|
@ -30,8 +30,6 @@ import math
|
|||
from ppcls.utils.save_load import load_dygraph_pretrain, load_dygraph_pretrain_from_url
|
||||
|
||||
MODEL_URLS = {
|
||||
"SE_ResNeXt50_vd_32x4d":
|
||||
"https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/SE_ResNeXt50_vd_32x4d_pretrained.pdparams",
|
||||
"SE_ResNeXt50_vd_32x4d":
|
||||
"https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/SE_ResNeXt50_vd_32x4d_pretrained.pdparams",
|
||||
"SENet154_vd":
|
||||
|
@ -304,16 +302,6 @@ def SE_ResNeXt50_vd_32x4d(pretrained=False, use_ssld=False, **kwargs):
|
|||
return model
|
||||
|
||||
|
||||
def SE_ResNeXt101_vd_32x4d(pretrained=False, use_ssld=False, **kwargs):
|
||||
model = ResNeXt(layers=101, cardinality=32, **kwargs)
|
||||
_load_pretrained(
|
||||
pretrained,
|
||||
model,
|
||||
MODEL_URLS["SE_ResNeXt101_vd_32x4d"],
|
||||
use_ssld=use_ssld)
|
||||
return model
|
||||
|
||||
|
||||
def SENet154_vd(pretrained=False, use_ssld=False, **kwargs):
|
||||
model = ResNeXt(layers=152, cardinality=64, **kwargs)
|
||||
_load_pretrained(
|
||||
|
|
Loading…
Reference in New Issue