From 091dcdfba2fffa79bef973f8c04f25b9d784fd4d Mon Sep 17 00:00:00 2001 From: John Bauer Date: Sun, 24 Dec 2023 01:22:41 -0800 Subject: [PATCH] Indentation error? --- .../lemma_classifier/transformer_baseline/baseline_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stanza/models/lemma_classifier/transformer_baseline/baseline_trainer.py b/stanza/models/lemma_classifier/transformer_baseline/baseline_trainer.py index 05091043a1..847ae383e8 100644 --- 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)