Fixed a bug in the computing outputs training.