Update efficientnet_transformer_glam.py

efficientnet_transformer_glam.py

@@ -200,8 +200,8 @@ class EfficientNetb0_TransformerGLAM(nn.Module):
 
         # Final Fusion
         F_out = g * T_out + (1 - g) * G_out
-        F_out = self.final_feature_layer(F_out)  # ✅ Final feature map for Grad-CAM
+        # ✅ Save the spatial feature map
+        self.last_feature = F_out
         pooled = F.adaptive_avg_pool2d(F_out, (1, 1)).view(B, -1)
-
         return self.fc(pooled)