2. 科技
    3. 【踩坑】RuntimeError: CUDA error: device-side assert triggered

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

    科技2022-07-11  81

    找了半天发现自己手动把原本代码里的sigmoid手动改成relu触发这个错误。

    score = score.sigmoid() # 我改成了 score = score.relu()
    Processed: 0.030, SQL: 8