pytorch报错RuntimeError: cuda runtime error (59) : device-side assert triggered at的奇怪解决方案
模型中添加了这样的函数
def func1(self,seq1,seq2):
out = []
//...往out里面加内容
return torch.stack(out,0)
运行报错RuntimeError: cuda runtime error (59) : device-side assert triggered at,位置在return之...
原创
2020-03-23 01:37:54 ·
1115 阅读 ·
1 评论