Update glam_efficientnet_model.py

glam_efficientnet_model.py

@@ -6,6 +6,7 @@ from typing import Optional, Union
 
 from glam_module import GLAM
 from swin_module import SwinWindowAttention
+from transformers import EfficientNetModel
 
 
 class GLAMEfficientNetConfig:
@@ -34,7 +35,8 @@ class GLAMEfficientNetForClassification(nn.Module):
 
         # ✅ 1) Torchvision EfficientNet Backbone
         efficientnet = models.efficientnet_b0(pretrained=False)  # No Hugging Face!
-        self.features = efficientnet.features
+        self.features = EfficientNetModel.from_pretrained("google/efficientnet-b0")
+
 
         # ✅ 1x1 conv for channel adjustment
         self.conv1x1 = nn.Conv2d(1280, config.embed_dim, kernel_size=1)