Hello All, I’m using vgg_16 model and i’m modifying the classifier to have two outputs.
(classifier): Sequential(
(0): Linear(in_features=25088, out_features=4096, bias=True)
(1): ReLU(inplace)
(2): Dropout(p=0.5)
(3): Linear(in_features=4096, out_features=4096, bias=True)
(4): ReLU(inplace)
(5): Dropout(p=0.5)
(6): Linear(in_features=4096, out_features=10, bias=True)
(7): Linear(in_features=4096, out_features=10, bias=True)
)
when i directly pass the input i get an of error size mismatch, can anyone help me how should i pass the input in order to get two outputs for a same images.