[pytorch][zero_grad]pytorch报错‘builtin_function_or_method‘ object has no attribute ‘named_modules‘
pytorch做grad-cam模型梯度归零时调用了zero_grad这个函数,然后报错发现原因在于torch.nn.ReLU函数在eval状态下会变成一个build_in_method的类,zero_grad的源码如下def zero_grad(self, set_to_none: bool = False) -> None: if getattr(self, '_is_replica', False): warnings.warn(
原创
2021-01-31 10:52:17 ·
1046 阅读 ·
0 评论