-
Notifications
You must be signed in to change notification settings - Fork 1.5k
Question about TorchVisionFC model #4867
Copy link
Copy link
Closed
Description
Discussed in #4865
Originally posted by ctestagrose August 8, 2022
I am interested in using the TorchVisionFC model. I see that the default model is resnet. Is there any way to use another torchvision model that is specified https://pytorch.org/vision/stable/models.html ? I would like to replace resnet with inception v3.
When I replace resnet with inception v3 I am receiving the following error:
RuntimeError: Expected 4-dimensional input for 4-dimensional weight [192, 768, 1, 1], but got 2-dimensional input of size [16, 1000] instead
I am thinking this is an issue with either the way I am loading the data or possibly a misinterpretation on my part in regard to the way the TorchVisionFC model works.
Thank you, I am extremely appreciative of any help.
Reactions are currently unavailable