PaddleClas
PaddleClas copied to clipboard
How to load pdparams file
- PaddleClas版本:PaddleClas release/2.5
- 训练环境信息:飞桨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))
等异常信息
想请教一下如何加载类似ResNet50_vd_ssld_v2这种不在内置模型库之内的模型来进行训练
这两个网络结构不一致,如果使用paddleclas提供的权重就只能通过paddleclas训练,否则需要手动转换