added mean KL divergence to log output