一个奇怪的BUG
解决办法:在模型加载后面加入一行代码 matplotlib.use( TkAgg’),这个bug的问题就是模型加载改变了matplotlib使用的终端,但为什么会改变,还是不知道啊
import torch
import thop
import torch.nn.utils .prune as prune
import matplotlib
import matplotlib.pyplot as plt
model = torch.load("runs/train/exp6/weights/last.pt")
matplotlib.use( TkAgg')
x = [i for i in range(3)]
y = [i for i in range(3)]
plt.plot(y,x)
plt.show()