调试yolov5程序的时候报错,torch.nn.modules.module.ModuleAttributeError: ‘Hardswish‘ object has no attribute ‘inplace‘
本机Anaconda3环境
![](https://img.haomeiwen.com/i14216194/de95a4ec4bc104f1.png)
解决办法
根据网上的方式,通过修改torch的代码成功解决问题。
找到环境路径
进入在Anaconda3中创建的环境,如:
![](https://img.haomeiwen.com/i14216194/b17df67ec0d8efca.png)
找到对应文件
找到activation文件,如图:
![](https://img.haomeiwen.com/i14216194/284a3258b6925ce7.png)
修改代码
在438行,删除self.inplace
def forward(self, input: Tensor) -> Tensor:
# return F.hardswish(input, self.inplace)
return F.hardswish(input)
![](https://img.haomeiwen.com/i14216194/b992513b44f7c7a7.png)
重新运行
关键环境,关键anaconda3,重新运行。
网友评论