Indentation error?

stanfordnlp · Dec 24, 2023 · 091dcdf · 091dcdf
1 parent 71be815
commit 091dcdf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/stanza/models/lemma_classifier/transformer_baseline/baseline_trainer.py b/stanza/models/lemma_classifier/transformer_baseline/baseline_trainer.py
@@ -64,7 +64,7 @@ def configure_weighted_loss(self, label_decoder: Mapping, counts: Mapping):
         total_samples = sum(counts.values())
         for class_idx in counts:
             weights[class_idx] = total_samples / (counts[class_idx] * len(counts))  # weight_i = total / (# examples in class i * num classes)
-            weights = torch.tensor(weights)
+        weights = torch.tensor(weights)
         logging.info(f"Using weights {weights} for weighted loss.")
         self.criterion = nn.BCEWithLogitsLoss(weight=weights)