bug fix

model_utils/efficientnet_config.py

@@ -292,21 +292,7 @@ class EfficientNetConfig(PretrainedConfig):
         self.depth_mult=depth_mult
                                     
         super().__init__(**kwargs)
-        
-        
-    def _init_weights(self, module):
-        # initialize weights before loading
-        # not all will have weights and biases
-        try:
-            module.weight.data.normal_(mean=0.0)
-        except AttributeError:
-            _ = None
-        try:
-            module.bias.data.zero_()
-        except AttributeError:
-            _ = None
-            
-        
+               
     
 class EfficientNetPreTrained(PreTrainedModel):
     
@@ -328,6 +314,19 @@ class EfficientNetPreTrained(PreTrainedModel):
     def forward(self, tensor):
         return self.model.forward(tensor)
     
+
+    def _init_weights(self, module):
+        # initialize weights before loading
+        # not all will have weights and biases
+        try:
+            module.weight.data.normal_(mean=0.0)
+        except AttributeError:
+            _ = None
+        try:
+            module.bias.data.zero_()
+        except AttributeError:
+            _ = None
+    
     
 class EfficientNet(nn.Module):