Search code examples
python-3.xpytorchhuggingface

Timm provides petrained models. On which dataset these weights are trained on?


How to find out on what dataset the petrained models provided from Timm are trained on?

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

https://timm.fast.ai

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


Solution

  • It's trained on ImageNet1k datasets. However, there are some models trained on the bigger dataset, Imagenet22k. Most of time, this is indicated in the name of the model.