Skip to content

Question about TorchVisionFC model #4867

@ctestagrose

Description

@ctestagrose

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.

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions