fixed log print bug due to softmax