Open muammar opened 5 years ago
Loading in a pre-trained model in PyTorch is simple:
from torchvision import models model = model.vgg16(pretrained=True)
Source: https://towardsdatascience.com/transfer-learning-with-convolutional-neural-networks-in-pytorch-dd09190245ce
I think the pretrained keyword argument is something that has to do with the vgg16 model but should not be complicated to implement.
pretrained
vgg16
Loading in a pre-trained model in PyTorch is simple:
Source: https://towardsdatascience.com/transfer-learning-with-convolutional-neural-networks-in-pytorch-dd09190245ce
I think the
pretrained
keyword argument is something that has to do with thevgg16
model but should not be complicated to implement.