import matplotlib
matplotlib.use('Agg')
# 使程序运行过程中不显示图形窗口
import matplotlib.pyplot as plt
trainHist, valHist = training.trainNetwork(...)
plt.xlabel('epoch')
plt.ylabel('loss')
plt.plot(valHist, label='validation loss')
plt.plot(trainHist, label='training loss')
plt.legend()
plt.savefig('loss.png')
Python 使程序运行过程中不显示图像窗口,而是将程序运行过程中的图像保存到本地
最新推荐文章于 2023-06-07 03:30:26 发布