Source code for EduNLP.ModelZoo.utils.downstream_output

import torch
from transformers.modeling_outputs import ModelOutput


[docs]class PropertyPredictionOutput(ModelOutput):
    loss: torch.FloatTensor = None
    logits: torch.FloatTensor = None


[docs]class KnowledgePredictionOutput(ModelOutput):
    loss: torch.FloatTensor = None
    logits: torch.FloatTensor = None