code: def inverse_decay_mix_prob(
      warmup_schedule_name, p_max, num_warmup_steps
  ):
      """Interpolate from 0.001 to 'p_max' over 'num_warmup_steps'."""
      warmup_schedule_fn = {
          "exp": common_layers.inverse_exp_decay,
          "linear": common_layers.inverse_lin_decay,
          "sigmoid": common_layers.inverse_sigmoid_decay,
      }[warmup_schedule_name]
      return p_max * warmup_schedule_fn(
          num_warmup_steps, min_value=0.001
      )
  
file_path: tensor2tensor/tensor2tensor/utils/scheduled_sampling.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import copy
  from tensor2tensor.layers import common_layers
  import tensorflow.compat.v1
  from tensorflow.compat.v1 import estimator
  from tensorflow.python.ops import inplace_ops
label_desc: Use image node_id 6 for calling a global function with example usage: inverse_decay_mix_prob(warmup_schedule_name, p_max, num_warmup_steps) and returns: unknown