Update modeling_protonet.py

modeling_protonet.py

@@ -854,7 +854,8 @@ class AudioProtoNetPreTrainedModel(PreTrainedModel):
             nn.init.trunc_normal_(module.weight, std=0.02)
             if module.bias is not None:
                 nn.init.zeros_(module.bias)
-        if isinstance(self.last_layer, LinearLayerWithoutNegativeConnections) and self.incorrect_class_connection is None:
+        if isinstance(module, LinearLayerWithoutNegativeConnections) and self.incorrect_class_connection is None:
+            # Initialize all weights to the correct_class_connection value
             self.last_layer.weight.data.fill_(self.correct_class_connection)
 
 
@@ -896,13 +897,6 @@ class AudioProtoNetForSequenceClassification(AudioProtoNetPreTrainedModel):
         self.model = AudioProtoNetModel(config)
         self.head = AudioProtoNetClassificationHead(config)
 
-
-    def freeze_backbone(self):
-        pass
-
-    def int2str(self): # TODO
-        pass
-
     def forward(
             self,
             input_values: torch.Tensor,