dhruv2842 commited on
Commit
f270b46
·
verified ·
1 Parent(s): bc30c66

Update efficientnet_transformer_glam.py

Browse files
Files changed (1) hide show
  1. efficientnet_transformer_glam.py +1 -1
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