当前位置: 代码迷 >> 综合 >> pytorch报错:RuntimeError: CUDA error: device-side assert triggered
  详细解决方案

pytorch报错:RuntimeError: CUDA error: device-side assert triggered

热度:78   发布时间:2023-12-17 17:14:24.0

模型输出的维度需要与类别数相等,比如

classifier = Classifier(INPUT_DIM, HIDDEN_DIM, OUTPUT_DIM).to(DEVICE)
logit = classifier(data)

data如果有5个类,那么OUTPUT_DIM也应该设置为5

  相关解决方案