一个奇怪的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()

原文地址:https://blog.csdn.net/m0_51977577/article/details/134663027

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任

如若转载,请注明出处:http://www.7code.cn/show_44022.html

如若内容造成侵权/违法违规/事实不符,请联系代码007邮箱suwngjj01@126.com进行投诉反馈,一经查实,立即删除

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注