ValueError: Arguments `target` and `output` must have the same shape. Received: target.shape=(None, 512), output.shape=(None, 3)
I was trying to train a bert model to solve a muti-classification problem:
I was trying to train a bert model to solve a muti-classification problem: