class_docstr: Compute the augmentation loss, as described in the
  `RangeAugment <https://arxiv.org/abs/2212.10553>`_ paper.
  
  Args:
      opts: command line arguments
class_docstr_tok: ["Compute","the","augmentation","loss",",","as","described","in","the","`","RangeAugment","<","https",":","\/\/arxiv.org\/abs\/2212.10553",">","`","_","paper",".","Args",":","opts",":","command","line","arguments"]
code: def add_arguments(
      cls, parser: argparse.ArgumentParser
  ) -> argparse.ArgumentParser:
      group = parser.add_argument_group(cls.__name__)
  
      group.add_argument(
          "--loss.neural-augmentation.perceptual-metric",
          type=str,
          default="psnr",
          choices=cls.__supported_metrics,
          help=f"Name of the perceptual metric to be used in {cls.__name__}.",
      )
  
      group.add_argument(
          "--loss.neural-augmentation.target-value",
          type=float,
          default=[40, 20],
          nargs="+",
          help=f"Target image similarity value in {cls.__name__}. Defaults to [40, 20]",
      )
  
      group.add_argument(
          "--loss.neural-augmentation.curriculum-method",
          type=str,
          default="cosine",
          choices=["linear", "cosine"],
          help=f"Curriculum for varying the target image similarity value in {cls.__name__}."
          f"Supported curriculums are {cls.__supported_metrics}. Defaults to cosine",
      )
      group.add_argument(
          "--loss.neural-augmentation.alpha",
          default=100.0,
          type=float,
          help="Scale loss value by alpha value. Defaults to 100. "
          "Note: When perceptual metric is PSNR, alpha value is divided by 65025",
      )
      return parser
  
file_path: ml-cvnets/loss_fn/neural_augmentation.py
imports: import argparse
  import math
  from typing import List, Mapping, Union
  import torch
  from torch import Tensor
  from torch.nn import functional
  from loss_fn import LOSS_REGISTRY, BaseCriteria
  from utils import logger
  from utils.ddp_utils import is_master
label_desc: Use image node_id 2 for calling the NeuralAugmentation obj's underlying member method code with example usage: obj.add_arguments(cls, parser) and returns: parser