PRUNE_EA parallel_search error about vega HOT 1 CLOSED

def calc_forward_latency(model, input, sess_config=None, num=100):
    """Model forward latency calculation.

    :param model: network model
    :type model: torch or tf module
    :param input: input tensor
    :type input: Tensor of torch or tf
    :param num: forward number
    :type num: int
    :return: forward latency
    :rtype: float
    """
    # step_cfg = UserConfig().data.get("nas")
    # if hasattr(step_cfg, "evaluator"):
    #     evaluate_cfg = step_cfg.get("evaluator")
    #     if hasattr(evaluate_cfg, "davinci_mobile_evaluator"):
    #         evaluate_config = evaluate_cfg.get("davinci_mobile_evaluator")
    #         latency = _calc_forward_latency_davinci(model, input, sess_config, evaluate_config)
    # else:
    latency = _calc_forward_latency_gpu(model, input, sess_config, num)
    return latency