def compute_avg_error(psnr, ssim, lpips):
  """The 'average' error used in the paper."""
  mse = psnr_to_mse(psnr)
  dssim = jnp.sqrt(1 - ssim)
  return jnp.exp(jnp.mean(jnp.log(jnp.array([mse, dssim, lpips]))))
The 'average' error used in the paper.
internal/math.py