【踩坑】RuntimeError: CUDA error: device-side assert triggered

今天NLP了嗎發表於2020-10-03

CUDA error: device-side assert triggered
找了半天發現自己手動把原本程式碼裡的sigmoid手動改成relu觸發這個錯誤。

score = score.sigmoid()
# 我改成了
score = score.relu()