- 博客(2)
- 收藏
- 关注
RSS订阅原创 解决RuntimeError: one of the variables needed for gradient computation has been modified by an inplace
在进行魔改模型得时候，出现了这个错误，上网查的说要把所有激活函数inplace改成False，试过之后发现是无效的，于是继续摸索，发现报错说的loss函数的变量被原地操作了，于是想到了先给计算出来的loss一个副本，也就是.clone()函数，完美解决！
2024-01-15 10:51:04
457
原创 在进行并行网络层的设计的时候，bottleneck模块参数问题
这样的相同的两个层，参数都是来源于layer3的两个层，在进行block的操作的时候会因为block共享了上一次的操作（即使是一样的结构在运行的时候也是顺序运行）的参数，导致参数报错，解决办法就是在这个makelayer的方法这里，重新定义一个新的方法，设置好inplanes。这就需要首先debug看正常层的通道数等等，确保通道数是一致的。
2024-01-13 19:37:07
418
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人