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

2023-12-23 18:16:29

随着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

文章来源:https://blog.csdn.net/a486259/article/details/135167190
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。