Plot W'bal on the fly in train mode

.. uses a fixed TAU of 300s so need to add some
   way of configuring it for on the fly calculations