Update efficientnet_transformer_glam.py
Browse files
efficientnet_transformer_glam.py
CHANGED
@@ -148,7 +148,7 @@ class EfficientNetb0_TransformerGLAM(nn.Module):
|
|
148 |
super(EfficientNetb0_TransformerGLAM, self).__init__()
|
149 |
|
150 |
# EfficientNet Backbone
|
151 |
-
efficientnet = models.efficientnet_b0(weights=None)
|
152 |
self.feature_extractor = nn.Sequential(*list(efficientnet.features.children()))
|
153 |
self.conv1x1 = nn.Conv2d(1280, embed_dim, kernel_size=1)
|
154 |
|
|
|
148 |
super(EfficientNetb0_TransformerGLAM, self).__init__()
|
149 |
|
150 |
# EfficientNet Backbone
|
151 |
+
efficientnet = models.efficientnet_b0(weights=None)
|
152 |
self.feature_extractor = nn.Sequential(*list(efficientnet.features.children()))
|
153 |
self.conv1x1 = nn.Conv2d(1280, embed_dim, kernel_size=1)
|
154 |
|