Reputation: 1
I am using pretrained vgg16 model import os
RuntimeError: Given groups=1, weight of size 64 3 3 3, expected input[1, 4, 120, 120] to have 3 channels, but got 4 channels instead
`
model = torchvision.models.vgg16(pretrained =True) `
i am unable to understand why error your text
Upvotes: 0
Views: 66