随着PaddleClas版本代码的迭代,博主以前的一些代码在使用上出现了bug,导致无法初始化模型,具体涉及paddle 42 将任意paddleclas模型作为paddledetection中的backbone使用代码的使用,为此重新对最新的PaddleClas代码进行梳理,实现重新初始化PaddleClas中的模型。
在迁移学习中应该初始化哪些模型可以参考如何选择出最适合的backbone模型?图像分类模型性能大摸底
一下代码就是仿照PaddleClas中engine.py代码中初始化模型的方式进行模型构建,调用get_model函数即可实现对任意PaddleClas中的模型进行初始化。
import paddle
import ppcls
import os
def load_dygraph_pretrain(model, path