Add z0callback option

Adds callback that calculates Z0 during the training to plot it alongside the loss

closed