plan.py 文件源码

def create(cls, mode):
    """Creates a plan.

    Args:
      mode: A string; 'train', 'eval', or 'infer'.

    Raises:
      ValueError: If `mode` is invalid.

    Returns:
      A Plan.
    """
    cases = {Plan.mode_keys.TRAIN: TrainPlan,
             Plan.mode_keys.EVAL: EvalPlan,
             Plan.mode_keys.INFER: InferPlan}
    if mode not in cases:
      raise ValueError('invalid mode %r not in %s' % (mode, sorted(cases)))
    return cases[mode]()