Open
Description
I can use the include_top
parameter in EfficientNet.from_name
, but in EfficientNet.from_pretrained
, I cannot use that. So, if I want to change the last layer of the pretrained model by self.base_model = nn.Sequential(*list(base_model.children())[:-1])
, it has the error in forward function.
Metadata
Metadata
Assignees
Labels
No labels