自己的情况是代码中有过多的图片显示，导致每个iteration/epoch训练时间越来越长。或其他类似的东西。

解决方法：

用plt.close()及时关闭图片

fig = plt.figure(0)  # 新图0
plt.savefig( )  # 保存
plt.close(0)  # 关闭图0

plt.close('all')  #关闭所有图

训练过程耗时截图：

 

使用 plt.close('all') 后