Python 使程序运行过程中不显示图像窗口，而是将程序运行过程中的图像保存到本地

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')