dhruv2842 commited on
Commit
ddbd5c2
Β·
verified Β·
1 Parent(s): da123ac

Update glam_efficientnet_model.py

Browse files
Files changed (1) hide show
  1. glam_efficientnet_model.py +1 -1
glam_efficientnet_model.py CHANGED
@@ -35,7 +35,7 @@ class GLAMEfficientNetForClassification(nn.Module):
35
 
36
  # βœ… 1) Torchvision EfficientNet Backbone
37
  efficientnet = models.efficientnet_b0(pretrained=False) # No Hugging Face!
38
- self.features = EfficientNetModel.from_pretrained("google/efficientnet-b0")
39
 
40
 
41
  # βœ… 1x1 conv for channel adjustment
 
35
 
36
  # βœ… 1) Torchvision EfficientNet Backbone
37
  efficientnet = models.efficientnet_b0(pretrained=False) # No Hugging Face!
38
+ self.feature_extractor = EfficientNetModel.from_pretrained("google/efficientnet-b0")
39
 
40
 
41
  # βœ… 1x1 conv for channel adjustment