I\'m fairly new with Python, and matplotlib, and using them to train neural networks. Now experimenting with VAE, and trying to keep track of its training by plotting means