I am trying to use the pre-trained models in pytorch and I am encountering problems with wide_resnet.
The part of the code that is suposed to access it is:
self.featEx = torchModels.resnext101_32x8d(pretrained=True)
self.featEx = torchModels.wide_resnet101_2(pretrained=True)
I have managed to get all the other models to work (I have left in the code for resnext as an example), but when I try to use it with the wide_resnet I get:
AttributeError: module ‘torchvision.models’ has no attribute ‘wide_resnet101_2’
I am thinking this is possibly a pytorch version problem but have been trying to update to the latest version without apparent success.