【Pytorch】 使用pytroch可视化工具时报错

报错信息：

RuntimeError: Output 0 of UnbindBackward is a view and is being modified inplace. This view is the output of a function that returns multiple views. Such functions do not allow the output views to be modified inplace. You should replace the inplace operation by an out-of-place one.

 

 

本地环境

pytorch 1.7
torchvision 0.8.0
python 3.8

由于 pytorch 1.7其中有一个错误具体详见：

https://github.com/pytorch/vision/issues/3025