本站消息

站长简介/公众号


站长简介:逗比程序员,理工宅男,前每日优鲜python全栈开发工程师,利用周末时间开发出本站,欢迎关注我的微信公众号:幽默盒子,一个专注于搞笑,分享快乐的公众号

  价值13000svip视频教程,python大神匠心打造,零基础python开发工程师视频教程全套,基础+进阶+项目实战,包含课件和源码

  出租广告位,需要合作请联系站长

+关注
已关注

分类  

暂无分类

标签  

暂无标签

日期归档  

2020-07(4)

2020-08(27)

nvidia xavier安装完torch1.7后出现Hardswish‘ object has no attribute ‘inplace‘

发布于2021-01-29 12:09     阅读(1293)     评论(0)     点赞(5)     收藏(2)



cuda10.2的nvidia xavier安装完torch1.7后,跑yolov5的detect文件,出现如下错误:

torch.nn.modules.module.ModuleAttributeError: ‘Hardswish‘ object has no attribute ‘inplace‘

有的博客提到了降低torch以及torchvision的版本,但是好不容易安装好的torch,我不太想重来,所以就采用了另一种方法:

conda activate yolov5_env
python
import torch
print(torch.__path__)

找到了torch的路径后,编辑/torch/nn/modules/activation.py,在438把“,self.inplace”删掉。

def forward(self, input: Tensor) -> Tensor:
    #return F.hardswish(input, self.inplace)  
    return F.hardswish(input)

然后再次执行就不会报错了。






所属网站分类: 技术文章 > 博客

作者:小鬼来了

链接:https://www.pythonheidong.com/blog/article/804072/d405899e21747b65513f/

来源:python黑洞网

任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任

5 0
收藏该文
已收藏

评论内容:(最多支持255个字符)