一堆bug
yeeanna
这个作者很懒,什么都没留下…
展开
-
RuntimeError: CUDA error: device-side assert triggered
说明问提出在s_output和label_source上。查了查unable to get repr for class说是数据对不上,才发现s_output的输出维度和数据集的类别数对不上,于是改了输出类别数num_class。省流:网络输出类别数和数据集实际类别数不相等。在改代码时经常遇到这个错误,于是记录下来。原创 2023-02-19 15:50:53 · 503 阅读 · 0 评论 -
AssertionError: Invalid device id
在执行如下代码: self.Model = torch.nn.DataParallel(self.Model.cuda(L.ids[0]), device_ids=L.ids) 出现这个错误: AssertionError: Invalid device id debud之后发现原因在于GPU只有一个’0’,而代码中L的id有’0’,'1’两块卡,于是把L.id改成只有0,解决问题. ...原创 2021-10-27 19:48:04 · 2479 阅读 · 0 评论