关于预训练模型参数的下载
来源:6-1 图像分类网络模型框架解读(上)

Sean_007
2021-04-23
你好,最近我有一个问题想要咨询您一下。请问pytorch官方预训练好的模型在哪里下载啊,有没有那种可以查询的网站或者资源?
比如说下图是我在网上搜索到的别人整理的,但是这个明显不全,比如我想下载一个faster rcnn的预训练模型参数,这里面就没有。(另,试图登陆网址https://download.pytorch.org/models/没有成功)
写回答
1回答
-
会写代码的好厨师
2021-05-06
预训练模型分成两种,一种是只有backbone.参数的模型,也就是你这里列出来的这些链接,这些模型现在不需要翻墙。另外,就是特定模型的预训练模型,这个要看你的repo是否有提供,如果没有提供是找不到预训练模型的,这种情况,只能用第一种,可能需要写一段加载参数的代码。建议,如果没有预训练模型,可以不使用。预训练模型在样本比较少的时候作用比较大,如果样本量足够,其实不需要预训练也一样的。
00
相似问题