Done training for output layer