关于Python的第三方库Matplotlib中plt.ion() & plt.ioff() & plt.savefig()函数的使用方法记录

---

版权声明：本文为博主原创文章，如需转载请贴上原博文链接：https://blog.csdn.net/u011628215/article/details/108379046

---

前言：最近又把莫烦老师的PyTorch教学视频学习一遍，当使用PyTorch做回归练习的时候，想把整个神经网络训练做成模块（莫烦老师的教程中，只将建立神经网络定义了类——class Net(torch.nn.Module)，我把训练过程、参数回传、优化器选择以及可视化分别做了函数），供以后直接调用，但是当做到可视化的时候，就遇到了问题，而问题就出在plt.ion()和plt.ioff()上。

注：from matplotlib import pyplot as plt

 

问题：莫烦老师的代码因为方便演示写成了一个整体（没有写成if __name__ == '__main__':的模块形式），所以运行没有问题；而当把代码做成单个的函数进行调用的时候，出现了可视化不连续的情况，即需要手动关闭figure窗口才能显示下一次训练的结果，如图1所示；