使用from torchvision.utils import save_image保存图像时,出现错误:
File “/usr/local/lib/python2.7/dist-packages/torchvision/utils.py”, line 51, in save_image
grid = make_grid(tensor, nrow=nrow, padding=padding)
File “/usr/local/lib/python2.7/dist-packages/torchvision/utils.py”, line 38, in make_grid
.narrow(2, x*width+1+padding//2, width-padding)\
RuntimeError: invalid argument 4: out of range at /pytorch/torch/lib/TH/generic/THTensor.c:439
原因是torchvision版本不对,解决方法是卸载torchvision,并重新安装,
pip uninstall torchvision
pip install torchvision