如何找出Timm提供的petrained模型是在什么数据集上训练的?
pretrained_resnet_34 = timm.create_model('resnet34', pretrained=True)
https://timm.fast.ai
https://github.com/huggingface/pytorch-image-models#models
如何找出Timm提供的petrained模型是在什么数据集上训练的?
pretrained_resnet_34 = timm.create_model('resnet34', pretrained=True)
https://timm.fast.ai
https://github.com/huggingface/pytorch-image-models#models
2条答案
按热度按时间xzv2uavs1#
它是在
ImageNet1k
数据集上训练的。然而,有一些模型是在更大的数据集Imagenet22k
上训练的。大多数情况下,这在模型的名称中表示。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