есть ошибка времени выполнения cuda (59), когда я тренирую свою модель с pytorch

Question

во время тренировки моей модели с pytorch возникает ошибка cuda (59). Я получаю ошибку как: «утверждение 'THCNumerics :: ge (val, zero)' не выполнено»

Я получаю ошибку времени выполнения (59), когда я тренирую модель, вот таккартина.Проблема появилась, когда я использую 128000 единиц данных для обучения, но не появилась, когда я использовал 12800 единиц данных для обучения. Кроме того, я использовал DataLoader для загрузки 128 частей данных на пакет. Код, который запрашивает проблемуis:

  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:" это код ошибки