trying to initialize all weights

app.py

@@ -117,11 +117,11 @@ def get_activations(model, image: list, model_name: str,
     
     
     layer_outputs = {}
-    for i in range(len(model.features)):
-        image = model.features[i](image)
+    for i in range(len(model.model.features)):
+        image = model.model.features[i](image)
         layer_outputs[i] = image
         print(i, layer_outputs[i].shape)
-    output = model(image).detach().cpu().numpy()
+    output = model.model(image).detach().cpu().numpy()
     output_1 = activation_indices[model_name].detach().cpu().numpy()
     output_2 = activation_indices[model_name].detach().cpu().numpy()
     

model_utils/efficientnet_config.py

@@ -303,7 +303,7 @@ class EfficientNetPreTrained(PreTrainedModel):
         config
     ):
         super().__init__(config)   
-        self.model = EfficientNet(  dropout=config.dropout,
+        self.model = EfficientNet(dropout=config.dropout,
                                     num_channels=config.num_channels,
                                     num_classes=config.num_classes,
                                     size=config.size,
@@ -320,8 +320,11 @@ class EfficientNetPreTrained(PreTrainedModel):
         # not all will have weights and biases
         try:
             module.weight.data.normal_(mean=0.0)
-        except AttributeError:
-            _ = None
+        except Exception:
+            try:
+                module.weight.data.fill_(1.0)
+            except Exception:
+                _ = None
         try:
             module.bias.data.zero_()
         except AttributeError: