Cannot handle this data type: (1, 1, 3000), <f4
tensor保存图片
import torchvision
torchvision.utils.save_image(output, './outputcnnpool/result333.png')
解决办法:
return 该变量以后
从函数里面移动到主函数里面
保存图片
Cannot handle this data type: (1, 1, 3000), <f4
tensor保存图片
import torchvision
torchvision.utils.save_image(output, './outputcnnpool/result333.png')
解决办法:
return 该变量以后
从函数里面移动到主函数里面
保存图片