write log probs instead of probs