2021-05-22

pytorch调试

刚开始像正常一样用gpu调试，报错：RuntimeError: CUDA error: device-side assert triggered，只有前面有数组越界的错误，后面没了。

choose_bg_inds =torch.tensor(np.random.choice((cls_inds_cpu).size()[0]), size=(choose_bg_num), replace=False),
                                              device=device)

刚开始看了别人的博客把cls_inds_cpu改成range(len(cls_inds_cpu))，这行代码不报错，后面一行不同的继续报错，弄来弄去，还是认真把这个方法理解了再改。
看别人的文章，改为用cpu调试，但是还是不能debug，只能自己写一个测试代码块，后面改了半天，发现可能是版本问题，后面有些代码维度根本对不上。