Warning: Grad strides do not match bucket view strides pytorch利用DDP报错

遇到报错: [W reducer.cpp:362] Warning: Grad strides do not match bucket view strides. This may indicate grad was not created according to the gradient layout contract, or that the param’s strides changed since DDP was constructed. This is not an error, but may impair performance.

机翻:警告。梯度与桶状视图的梯度不一致。这可能表明grad没有按照梯度布局合同创建,或者参数的步长在DDP构建后发生了变化。 这不是一个错误,但可能影响性能。

方法1:
Github的issue:Grad strides do not match bucket view strides.#47163
pytorch利用DDP进行加速的报错问题
在transpose或者permute后面加 .contiguous()
但是加了之后还是有警告
在这描述

方法2:
类似于x = x[:, :H, :W, :]这种切片的操作
改成:

x = x[:, :H, :W, :].contiguous()

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
乘风的头像乘风管理团队
上一篇 2023年8月17日
下一篇 2023年8月17日

相关推荐