![](https://img-blog.csdnimg.cn/20201014180756918.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
等待小弟666
这个作者很懒,什么都没留下…
展开
-
pytorch中trace时出现forward() takes 2 positional arguments but 3 were given问题
错误源代码import torchclass MyCell(torch.nn.Module): def __init__(self): super(MyCell, self).__init__() self.linear = torch.nn.Linear(4, 4) def forward(self, input): x = input[0] h = input[1] new_h = torch.tan原创 2021-03-20 21:58:03 · 17764 阅读 · 2 评论 -
利用libtorch在C++程序上部署pytorch模型
在torch上导出模型,新建ResNet50.pyimport torchimport torchvision.models as modelsfrom PIL import Imageimport numpy as npimage = Image.open("0.jpg") #图片发在了build文件夹下image = image.resize((224, 224),Image.ANTIALIAS)image = np.asarray(image)image = image / 255i原创 2021-02-13 20:35:08 · 330 阅读 · 1 评论 -
FewShot目标检测
小样本学习:https://blog.csdn.net/weixin_37589575/article/details/92801610?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-2.control&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-2.controlFewShot目标检测20原创 2020-12-10 19:54:05 · 549 阅读 · 0 评论 -
安装Paddle出现Cannot load cudnn shared library问题
参考链接:https://github.com/PaddlePaddle/Paddle/issues/22996安装Paddle1.8.3之后,出现问题:解决方案:(1)对应cuda版本,安装对应的cudnn旧cudnn版本下载位置:https://developer.nvidia.com/rdp/cudnn-archive(2)解压之后,将cudnn文件拷贝至cuda10.1(我使用的cuda版本是10.1)中sudo cp ~/Downloads/cuda/include/cudnn.h原创 2020-08-25 08:40:58 · 2956 阅读 · 0 评论