Traceback (most recent call last):


 "train_fast_bert_doc_rerank.py", line 81, in <module>

optimizer_type="lamb"


 "/usr/local/lib/python3.6/site-packages/fast_bert/learner_cls.py", line 295, in fit

results = self.validate()


 "/usr/local/lib/python3.6/site-packages/fast_bert/learner_cls.py", line 382, in validate

validation_scores[metric['name']] = metric['function'](all_logits, all_labels)


 "/usr/local/lib/python3.6/site-packages/fast_bert/metrics.py", line 31, in accuracy_thresh

return ((y_pred > thresh) == y_true.byte()).float().mean().item()
RuntimeError: The size of tensor a (2) must match the size of tensor b (9833) at non-singleton dimension

Traceback (most recent call last):


 "/mnt/home/dunan/Learn/Kaggle/google_quest/kaggle_google_quest/train.py", line 267, in <module>

train(args)


 "/mnt/home/dunan/Learn/Kaggle/google_quest/kaggle_google_quest/train.py", line 138, in train

score, val_loss = predict(model, val_loader)


 "/mnt/home/dunan/Learn/Kaggle/google_quest/kaggle_google_quest/train.py", line 252, in predict

preds.append(outputs.cpu().numpy())
RuntimeError: CUDA error: device-side assert triggered
