PaddleClas icon indicating copy to clipboard operation
PaddleClas copied to clipboard

How to load pdparams file

Open SaitamaSiSi opened this issue 1 year ago • 1 comments

  1. PaddleClas版本:PaddleClas release/2.5
  2. 训练环境信息:飞桨AI Studio

network = paddle.vision.models.resnet50(num_classes=get('num_classes'), pretrained=True) model = paddle.Model(network) ... model.fit(...)

我在飞桨AI Studio中通过上述代码进行模型的加载和训练,可我发现ResNet50_vd_ssld_v2模型并未在paddle.vision.models中,于是我下载了ResNet50_vd_ssld_v2_pretrained.pdparams文件,想通过以下代码方法进行加载

layer_state_dict = paddle.load("./work/ResNet50_vd_ssld_v2_pretrained.pdparams") network.load_dict(layer_state_dict)

但好像并未成功,提示 warnings.warn(f"Skip loading for {key}. " + str(err)) 等异常信息 3745EAFE-BBCC-4bca-952E-140C4874AA05

想请教一下如何加载类似ResNet50_vd_ssld_v2这种不在内置模型库之内的模型来进行训练

SaitamaSiSi avatar Sep 21 '23 06:09 SaitamaSiSi

这两个网络结构不一致,如果使用paddleclas提供的权重就只能通过paddleclas训练,否则需要手动转换

cuicheng01 avatar Oct 10 '23 11:10 cuicheng01