是自身输入的图片应该是112112的,结果误输入了原图分辨率为540640,所以报cuda out of memory的问题,修改后,显存够用不再报错,另外可以加大pycharm的内存
详见:https://blog.csdn.net/weixin_41010198/article/details/89948866
是自身输入的图片应该是112112的,结果误输入了原图分辨率为540640,所以报cuda out of memory的问题,修改后,显存够用不再报错,另外可以加大pycharm的内存
详见:https://blog.csdn.net/weixin_41010198/article/details/89948866