我使用的是libtorch1.5.1 ,发现当动态图没有达到设定的最大长边时,只要新的尺寸比之前的最大尺寸大就会增加一部分显存,最终的显存会比最大尺寸大很多。
自己的解决办法是在推理第一张图之前先推理一张最大尺寸的空图。
pyotrch最新版本好像解决了这个问题,不过是pytorch1.8版本解决的,pytorch1.7还没有解决。
我使用的是libtorch1.5.1 ,发现当动态图没有达到设定的最大长边时,只要新的尺寸比之前的最大尺寸大就会增加一部分显存,最终的显存会比最大尺寸大很多。
自己的解决办法是在推理第一张图之前先推理一张最大尺寸的空图。
pyotrch最新版本好像解决了这个问题,不过是pytorch1.8版本解决的,pytorch1.7还没有解决。