2021-04-23

语义分割SPNet文章代码复现之解决bug

1.出现权重参数和模型结构参数不匹配的问题

这个错误我真的找了好久，一直对照模型参数结构来看，怎么也没找到错误，之前一直没有怀疑权重文件的问题，后来实在没办法了，换了官方的权重文件（刚开始用别人给的权重文件是为了省时间），终于解决了！错误如下

2.出现CUDA out ofmemory的错误，网上找了很多办法，最多的是让你加with torch.no_grad，但是我不确定是在哪加，最后把batch size改小就解决这个问题了。