paddle 54 从PaddleClas2.5初始化模型用于迁移学习(LeViT、ReXNet、EfficientNet等)

随着PaddleClas版本代码的迭代,博主以前的一些代码在使用上出现了bug,导致无法初始化模型,具体涉及paddle 42 将任意paddleclas模型作为paddledetection中的backbone使用代码的使用,为此重新对最新的PaddleClas代码进行梳理,实现重新初始化PaddleClas中的模型。

在迁移学习中应该初始化哪些模型可以参考如何选择出最适合的backbone模型?图像分类模型性能大摸底

1、完整代码

一下代码就是仿照PaddleClas中engine.py代码中初始化模型的方式进行模型构建,调用get_model函数即可实现对任意PaddleClas中的模型进行初始化。

import paddle
import ppcls
import os
def load_dygraph_pretrain(model, path

你可能感兴趣的:(paddlepaddle,paddle,迁移学习,PaddleClas,深度学习)