pytorch Timm提供petrained模型,这些权重是在哪个数据集上训练的?

kdfy810k  于 2023-05-07  发布在  其他
关注(0)|答案(2)|浏览(180)

如何找出Timm提供的petrained模型是在什么数据集上训练的?

pretrained_resnet_34 = timm.create_model('resnet34', pretrained=True)

https://timm.fast.ai
https://github.com/huggingface/pytorch-image-models#models

xzv2uavs

xzv2uavs1#

它是在ImageNet1k数据集上训练的。然而,有一些模型是在更大的数据集Imagenet22k上训练的。大多数情况下,这在模型的名称中表示。

yeotifhr

yeotifhr2#

ImageNet(https://image-net.org/download)大多数,可以通过模型名称来确定,在许可证中提到了https://github.com/huggingface/pytorch-image-models#licenses
作者指出,所有这些都是在ImageNet上预训练的,“这里包含的几个权重或引用都是用专有数据集预训练的”-https://github.com/huggingface/pytorch-image-models#licenses(2023年5月4日)。基本上,Google和FB在这里有模型(数据集没有与作者共享)您可以通过阅读相关的www.example.com文档使用此模型来更彻底地检查您使用的模型arxive.org:文档列表在这里https://github.com/huggingface/pytorch-image-models

相关问题