作者:岩蕃wy之人 | 来源:互联网 | 2023-06-07 16:22
I tried to remove top layers like follows:base_model = EfficientNet.from_pretrained('efficientnet-b3
I tried to remove top layers like follows:
base_model = EfficientNet.from_pretrained('efficientnet-b3')
model = nn.Sequential(*list(base_model.children()[:-3]))
x = torch.randn((1,3,300,300))
model(x)
It throws "NotImplementedError" for forward(). Is there a way to not include top layers?
该提问来源于开源项目:lukemelas/EfficientNet-PyTorch
is it going to be in the next pip version? Or do you know when it's planned for release?