pytorch - 当我使用 pytorch 训练模型时出现 cuda 运行时错误（59）

Question

当我用 pytorch 训练我的模型时出现 cuda 运行时错误（59）。我收到错误消息：“断言 'THCNumerics::ge(val,zero)' 失败”

当我训练模型时，我得到一个 cuda 运行时错误（59），就像那张照片一样。使用128000条数据进行训练时出现问题，使用12800条数据进行训练时没有出现。另外，我使用DataLoader每批次加载128条数据。提示问题的代码是 ：

  for i in range(batch_size):
            idxs_data = idxs[i].data
            for j in range(seq_len):
                if mask[i][j] == 1:
                    pass

“如果掩码[i][j] == 1:”是错误代码

标签： pytorch