def load_metric_params(config: ml_collections.ConfigDict) -> Dict[str, Any]:
  """Loads the auxiliary parameters to compute the metrics."""
  metric_params = {}
  if config.eval.get('enable_inception_score', False):
    metric_params['inception_score'] = inception_score.load_params(
        config.eval.get('inception_score_checkpoint_filename'))
  if config.eval.get('enable_frechet_distance', False):
    metric_params['frechet_distance'] = frechet_distance.load_params(
        config.eval.get('frechet_distance_checkpoint_filename'))
  if config.eval.get('enable_lpips', False):
    lpips_model_dict = image_quality_metrics.load_lpips_models()
    metric_functions = dict(
        lpips_alexnet=functools.partial(
            image_quality_metrics.lpips_tf,
            lpips_model=lpips_model_dict['lpips_alexnet']),
        lpips_vgg=functools.partial(
            image_quality_metrics.lpips_tf,
            lpips_model=lpips_model_dict['lpips_vgg']))
    metric_params['lpips'] = metric_functions
  return metric_params
Loads the auxiliary parameters to compute the metrics.
videogvt/train_lib/eval_utils.py