torchrecipes/text/doc_classification/module/doc_classification.py [117:128]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        self,
        batch: Mapping[str, torch.Tensor],
        batch_idx: int,
        *args: Any,
        **kwargs: Any,
    ) -> None:
        logits = self.model(batch["token_ids"])
        loss = self.loss(logits, batch["label_ids"])
        scores = F.softmax(logits)

        self.accuracy(scores, batch["label_ids"])
        self.fbeta(scores, batch["label_ids"])
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



torchrecipes/text/doc_classification/module/doc_classification.py [134:145]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        self,
        batch: Mapping[str, torch.Tensor],
        batch_idx: int,
        *args: Any,
        **kwargs: Any,
    ) -> None:
        logits = self.model(batch["token_ids"])
        loss = self.loss(logits, batch["label_ids"])
        scores = F.softmax(logits)

        self.accuracy(scores, batch["label_ids"])
        self.fbeta(scores, batch["label_ids"])
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -