pytorch | yolov5 Can not get arrribute SiLU

本文主要是介绍pytorch | yolov5 Can not get arrribute SiLU，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

【报错】

Can‘t get attribute ‘SiLU‘ on ＜module ‘torch.nn.modules.activation

【解决方法】

将SiLU拷贝到~/.local/lib/python3.8/site-packages/torch/nn/modules/activation.py文件中

class SiLU(Module):  # export-friendly version of nn.SiLU()@staticmethoddef forward(x):return x * torch.sigmoid(x)

这篇关于pytorch | yolov5 Can not get arrribute SiLU的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---