tensorflow/models · error · ValueError

Invalid augmentation_name: {}

Error message

Invalid augmentation_name: {}

What it means

Error "Invalid augmentation_name: {}" thrown in tensorflow/models.

Source

Thrown at official/legacy/image_classification/augment.py:849

        applied in parallel to a single image (2 vs 3).
      policies: list of lists of tuples in the form `(func, prob, level)`,
        `func` is a string name of the augmentation function, `prob` is the
        probability of applying the `func` operation, `level` is the input
        argument for `func`.
      cutout_const: multiplier for applying cutout.
      translate_const: multiplier for applying translation.
    """
    super(AutoAugment, self).__init__()

    if policies is None:
      self.available_policies = {
          'v0': self.policy_v0(),
          'test': self.policy_test(),
          'simple': self.policy_simple(),
      }

    if augmentation_name not in self.available_policies:
      raise ValueError(
          'Invalid augmentation_name: {}'.format(augmentation_name))

    self.augmentation_name = augmentation_name
    self.policies = self.available_policies[augmentation_name]
    self.cutout_const = float(cutout_const)
    self.translate_const = float(translate_const)

  def distort(self, image: tf.Tensor) -> tf.Tensor:
    """Applies the AutoAugment policy to `image`.

    AutoAugment is from the paper: https://arxiv.org/abs/1805.09501.

    Args:
      image: `Tensor` of shape [height, width, 3] representing an image.

    Returns:
      A version of image that now has data augmentation applied to it based on
      the `policies` pass into the function.

View on GitHub (pinned to e006f5f0d5)

When it happens

Trigger: Thrown at official/legacy/image_classification/augment.py:849 when the library encounters an invalid state.

Common situations: See trigger scenarios.

AI-assisted analysis of tensorflow/models@e006f5f0d5 (2026-08-24). Data as JSON: /api/errors/79c7f6704f592d4c. Report an issue: GitHub.