enable output stats while training

and cross check GPU generated one with CPU generated one