pytorch
sheepcyyc
这个作者很懒,什么都没留下…
展开
-
pytorch class ‘NoneType‘
建议看一下自己的网络结构 forward部分有没有return。没有return,网络返回类型是class 'NoneType'原创 2021-09-19 17:30:18 · 837 阅读 · 0 评论 -
pytorch 就地操作 显存节省
X = torch.arange(12, dtype=torch.float32).reshape((3, 4))Y = torch.tensor([[2.0, 1, 4, 3], [1, 2, 3, 4], [4, 3, 2, 1]])示例1before = id(Y)Y = Y + Xid(Y) == before以上代码输出结果为False。这表明进行矩阵加操作前的矩阵Y和进行操作后的矩阵Y本质上存储地址是不一样的。这主要是因为进行Y = Y + X操作时,pytorch会进行.原创 2021-04-13 17:37:52 · 272 阅读 · 0 评论 -
pytorch reshape view clone的区别
reshape view作为pytorch中torch的常用操作,有一些小细节需要注意一下。x = torch.arange(12)print('x')print(x)print(id(x))y = x.reshape(3,4)print('y')print(y)print(id(y))x[:] = 2print('x')print(x)print('y')print(y)以上代码输出结果为xtensor([ 0, 1, 2, 3, 4, 5, 6原创 2021-04-13 01:22:49 · 292 阅读 · 0 评论