General training updates
v0.3

Fix for predictions not using normalisation