亲测解决Grad strides do not match bucket view strides

狂小虎

已于 2024-11-22 09:09:10 修改

阅读量941

点赞数 9

CC 4.0 BY-SA版权

于 2024-11-22 09:08:36 首次发布

本文为博主原创文章，未经博主“狂小虎”本人允许不得转载。

本文链接：https://blog.csdn.net/Davidietop/article/details/143960918

这个问题是由于小虎在用DP的时候出现的，解决方法是在特别的变量后面加上.contiguous()。

环境

python 3.10 + Pytorch2.0

问题原文

/home/xiaohu/anaconda3/envs/smor/lib/python3.10/site-packages/torch/autograd/__init__.py:251: UserWarning: 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.sizes() = [3, 24, 1, 1], strides() = [24, 1