Epoch training finished, needs testing