class_docstr: Duration model of phonemes.
class_docstr_tok: ["Duration","model","of","phonemes","."]
code: def __init__(self, is_training=True):
      super().__init__()
      self.is_training = is_training
      self.encoder = TokenEncoder(
          FLAGS.vocab_size,
          FLAGS.duration_lstm_dim,
          FLAGS.duration_embed_dropout_rate,
          is_training,
      )
      self.projection = hk.Sequential(
          [
              hk.Linear(FLAGS.duration_lstm_dim),
              jax.nn.gelu,
              hk.Linear(1),
          ]
      )
  
imports: import haiku
  import jax
  import jax.numpy
  from jax.numpy import ndarray
  from .config import FLAGS, AcousticInput, DurationInput
label_desc: Use image node_id 1 to create a new DurationModel object from inherited base classes: hk with example: obj = DurationModel(is_training)