Try removing that permute and re-run?

Do you see this issue in the first iteration or later during the training?
Could you add these print statements and check the shapes?

print(batch.tweet.shape)
print(predictions.shape)
print(batch.label.shape)
acc = binary_accuracy(predictions, batch.label)

Add print statements in the forward to see yourself where it’s breaking;