Pytorch1.7报错 Output 0 of UnbindBackward is a view and is being modified inplace

更新pytorch1.7 之后运行程序,有

utils.make_grid()

报下面的错误。
在这里插入图片描述
进入utils.py文件,把74行改成,就好用了

for idx, _ in enumerate([jj for jj in tensor]):

在这里插入图片描述
参考 https://github.com/pytorch/vision/issues/3025

posted @ 2020-11-29 12:52  赫凯  阅读(314)  评论(0)    收藏  举报