Spaces:

bvd757
/

Papers_classification

Running

bvd757 commited on Apr 8

Commit

e15e482

1 Parent(s): a0a0073

app fixes_v4

Files changed (1) hide show

app.py CHANGED Viewed

@@ -57,7 +57,7 @@ class DebertPaperClassifier(torch.nn.Module):
             torch.nn.Linear(512, num_labels)
         )
-        self._init_weights()
         if class_weights is not None:
             self.loss_fct = torch.nn.BCEWithLogitsLoss(weight=class_weights.to(device))
         else:

             torch.nn.Linear(512, num_labels)
         )
+        #self._init_weights()
         if class_weights is not None:
             self.loss_fct = torch.nn.BCEWithLogitsLoss(weight=class_weights.to(device))
         else: