load_state_dict时候报了cuda out of memory

我加载的模型明明可以装进去但是显存溢出,更神奇的是比如我第一个python文件已经load好在跑了,第二个文件load同一个pth文件就不行了,更有甚者我第一个这时候再运行也不行了。

我不明白原因,但是有个规律可以参考:就是如果我第一个文件已经load进来这个pth正在跑,别的文件好像就不能再去load这个pth了(在跑完第一个前),即使我复制一份也不太行。。神奇,我就直接找了另一个pth拿过来试了试发现就可以了

你可能感兴趣的:(Debug之路)