Open dkoes opened 2 years ago
Trying to run the example colab notebook fails with the error AttributeError: 'EfficientNet' object has no attribute 'act1'
AttributeError: 'EfficientNet' object has no attribute 'act1'
This is because there is no act1 attribute in the model returned by timm.create_model('tf_efficientnet_lite0', pretrained=True)
act1
timm.create_model('tf_efficientnet_lite0', pretrained=True)
It is also the case that this model only has 7 blocks, yet _make_efficientnet seems to assume there's at least nine, which is concerning:
_make_efficientnet
def _make_efficientnet(model): pretrained = nn.Module() pretrained.layer0 = nn.Sequential(model.conv_stem, model.bn1, model.act1, *model.blocks[0:2]) pretrained.layer1 = nn.Sequential(*model.blocks[2:3]) pretrained.layer2 = nn.Sequential(*model.blocks[3:5]) pretrained.layer3 = nn.Sequential(*model.blocks[5:9]) return pretrained
!pip install timm==0.5.4 try this
Trying to run the example colab notebook fails with the error
AttributeError: 'EfficientNet' object has no attribute 'act1'
This is because there is no
act1
attribute in the model returned bytimm.create_model('tf_efficientnet_lite0', pretrained=True)
It is also the case that this model only has 7 blocks, yet
_make_efficientnet
seems to assume there's at least nine, which is concerning: