提交 d3e2f7a5 编写于 作者: C cuicheng01

update se_resnext_vd.py

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