项目场景:
提示:这里简述项目相关背景:
在进行利用pytorch进行yolov5目标检测训练的时候报错。
问题描述1:
yolov5报错代码如下: AttributeError: ‘Hardswish’ object has no attribute ‘inplace’
原因分析1:
主要是activation.py报错,定位到这一行代码为:
def forward(self, input: Tensor) -> Tensor:
return F.hardswish(input, self.inplace)
解决方案1:
原来代码为:
def forward(self, input: Tensor) -> Tensor:
return F.hardswish(input, self.inplace)
将代码修改为:
def forward(self, input: Tensor) -> Tensor:
return F.hardswish(input)
将self.inplace给删除掉
当我在尝试进行yolov5进行训练时,运行defect.py,又开始报了一种错误,具体报错内容如下:
AttributeError: ‘Upsample’ object has no attribute
‘recompute_scale_factor’
主要是upsamlping出现了问题,应该是上采样函数报错,定位到具体代码位置为:
原来代码为:
def forward(self, input: Tensor) -> Tensor:
return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners,
recompute_scale_factor=self.recompute_scale_factor)
修改成为:文章来源:https://www.toymoban.com/news/detail-617842.html
def forward(self, input: Tensor) -> Tensor:
return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners)
再次运行defect.py就可以预测了。文章来源地址https://www.toymoban.com/news/detail-617842.html
到了这里,关于【yolov5 defect报错】AttributeError: ‘Hardswish‘ object has no attribute ‘inplace‘的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!