Target size that is different to the input size

[yongen9696]

@XYBMA

...
class BODDetector(pl.LightningModule):

  def __init__(self, n_classes: int, n_training_steps=None, n_warmup_steps=None):
    super().__init__()
    self.bert = BertModel.from_pretrained(MODEL_NAME, return_dict=True)
    self.classifier = nn.Linear(self.bert.config.hidden_size, n_classes)
...

perhaps it is due to how you have defined your nn.Linear