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

Question

当我用pytorch训练模型时，有cuda运行时错误（59）。 I get the error as: "assertion ‘THCNumerics::ge(val,zero)’ failed"

训练模型时，我得到一个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

“ if mask [i] [j] == 1”是错误代码