前言: 本文主要记载在pytorch使用过程中遇到的一些报错以及解决方案等,在以前的文章[1]中,主要涉及的是比较容易出现的原理上的,或者难以发现的bug,而这里的主要是系统抛出的error或者warning的解决方法,其各有侧重,欢迎各位贡献idea。如有谬误,请联系指出,如需转载,请注明出处,谢谢。
联系方式:
e-mail: FesianXu@gmail.com
QQ: 973926198
github: https://github.com/FesianXu
list
。 解决方案:将所有需要优化的对象用list
进行包裹,如:trained_vars=list(model.submodel_1.parameters()) + \\
list(model.submodel_2.parameters())
opt=torch.optim.SGD(trained_vars, lr=...)
注意到此时的trained_vars
是一个可迭代的list
,其中每个元素都是一个参数组。
class model(nn.Module):
def __init__(self):
super().__init__()
self.convs=[
nn.Conv2d(...),
nn.Conv2d(...)
]
m=model().cuda()
此时convs
因为只是个list
而不是集成了nn.Module
的对象,因此是不会被迭代地放置到GPU中的,model().cuda()
这种方式只会对model
类中的__init__()
方法中定义的继承了nn.Module
的类的实例有效。 解决方法:
self.convs=[
nn.Conv2d(...),
nn.Conv2d(...)
]
self.convs=nn.ModuleList(self.convs)
tensor
的数据类型不匹配,应该是torch.float32()
解决方法:inputv=inputv.float()
model(inputv)
in-place
操作,比如x +=1
,x[0]=5
等,这类操作如果是在如下情况发生的:c=x+b
x +=1
compute_gradient(ys=c, xs=x)
如果此时c
对x
对梯度,而x
的值已经在后续的计算中发生了in-place
的变化,那么就会导致出错[5]
解决方案:将x +=1
改成x=x+1
即可解决。
补充: x +=1
在pytorch
中是一个in-place
操作,所谓in-place
就是进行了该操作是对该变量直接进行处理后返回的,也就是说该变量的地址是不会改变的,只是值变了而已;而非in-place
操作就是,在内存中另外开辟了一个空间存放新的值,然后将指针指向那个新的地址,在这种情况下,该变量的地址是会改变的。下面是一个例子:
In[1]: import torch
In[2]: a=torch.tensor(5)
In[3]: a
Out[3]: tensor(5)
In[4]: id(a)
Out[4]: 139636046877104
In[5]: a +=8
In[6]: a
Out[6]: tensor(13)
In[7]: id(a)
Out[7]: 139636046877104 # inplace操作,其地址不变
In[1]: import torch
In[2]: a=torch.tensor(6)
In[3]: id(a)
Out[3]: 139938915856816
In[4]: a=a+1
In[5]: a
Out[5]: tensor(7)
In[6]: id(a)
Out[6]: 139938915829584 # 非in-place操作,地址不同了
在pytorch中,计算梯度的时候有时候要求前面的相关变量不能被in-place
操作改变,不然将会导致梯度计算问题,从而报错。
forward
方法为非静态方法,在1.3版本之后,forward
方法要求定义成静态方法,具体参考[6]。torch.backends.cudnn.enabled=False
import torch
torch.backends.cudnn.enabled=False
# your code here
pytorch
有关的第三方包时,如果先导入第三方包,容易发生这种错误,正确的做法是首先导入pytorch
,例如:[7]import torch
import your_extension
[1]. 用pytorch踩过的坑[2]. https://discuss.pytorch.org/t/giving-multiple-parameters-in-optimizer/869[3]. https://pytorch.org/docs/stable/optim.html?highlight=sgd#torch.optim.ASGD[4]. https://blog.csdn.net/LoseInVain/article/details/81708474[5]. https://discuss.pytorch.org/t/encounter-the-runtimeerror-one-of-the-variables-needed-for-gradient-computation-has-been-modified-by-an-inplace-operation/836/5[6]. https://discuss.pytorch.org/t/custom-autograd-function-must-it-be-static/14980[7]. https://github.com/pytorch/extension-cpp/issues/6[8]. https://towardsdatascience.com/cuda-error-device-side-assert-triggered-c6ae1c8fa4c3[9]. https://github.com/pytorch/pytorch/issues/27588#issuecomment-603315518
我们的团队人数
我们服务过多少企业
我们服务过多少家庭
我们设计了多少方案