When we use pre-trained models from trochvision, we are using
model = models.vgg16(pretrained = True).
I am using a github repo of a network, which is not included in the torchvision. How can I assign the model?
model = XXXXXX(pretrained = True).
what should be this XXXXXX?
Any help would be appreciated. Sorry to ask a simple question,but I have been struggling with this.
Thanks