Hi @ptrblck,
thank you very much for your quick (and very helpful) response.
Just a quick question; I assume that you use the pretrained vgg16 model from import torchvision.models as models, right? If so, you load it as vgg = models.vgg16(pretrained=False), but since I need the pre-trained model, shouldn’t I use pretrained=True (btw, where is this pretrained? On ImageNet for 1000 classes maybe?).