未验证 提交 d618efe7 编写于 作者: C cuicheng01 提交者: GitHub

Merge pull request #1504 from cuicheng01/release/2.3

[cherry-pick]update se_resnext_vd.py
......@@ -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.
先完成此消息的编辑!
想要评论请 注册