pytorch
孙小l懒_keep_update
这个作者很懒,什么都没留下…
展开
-
RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS
我的PyTorch version 0.4.1,cudatoolkit 9.0cudnn 7.6.5在程序中插入下面这句话,很好地解决了!至于更深入的原因目前不是很清楚,其中解决方案来源于RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESStorch.backends.cudnn.benchmark = True...原创 2021-08-07 14:29:12 · 1715 阅读 · 3 评论 -
torch.nn.Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, b
torch.nn.Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True, padding_mode=‘zeros’)卷积核与原图按照左上角对齐的原则进行卷积操作stride:步幅,卷积核两次卷积左上角点之间的距离;通俗讲就是卷积核每隔几个像素进行一次...原创 2020-01-11 18:37:00 · 6392 阅读 · 0 评论 -
Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same出错原因是:网络的权重数据类型没有加载成torch.cuda.FloatTensor,但是训练用的数据已经在训练文件中转成了torch.cuda.FloatTensor类型。出错的原因是:自己编写的网络忘记nn.Mo...原创 2020-01-11 15:37:36 · 2055 阅读 · 0 评论 -
pytorch显卡号指定
pytorch显卡号默认选择01.torch.cuda.set_device(1)此法适合在生成网络对象之前使用,即放在最顶层的文件中,而且要import torch2.self.net_bone = self.net_bone.cuda(i)初始化模型时使用,但是要注意将数据存放的GPU也要修改,不然可能会报错,一般在程序中寻找sal_image, sal_label, sal_edge...原创 2020-01-06 22:00:59 · 1061 阅读 · 0 评论