Hey everyone. I have a question. Are the pre-trained models available in torch-vision.models trained using the imagenet dataset? If not, how can i get the ones trained on the imagenet dataset ?
Yes, a lot of pretrained
torchvision.models were pretrained on ImageNet as seen here. The docs also show how to access the pretrained weights which now also include the dataset name, e.g.