Method build_model

language/mentionmemory/tasks/base_task.py:37–40

Builds model by instantiating flax module associated with task.

(cls,
                  model_config: ml_collections.FrozenConfigDict)

  @classmethod
  def build_model(cls,
                  model_config: ml_collections.FrozenConfigDict) -> nn.Module:
    """Builds model by instantiating flax module associated with task."""
    return cls.model_class(**model_config)

  @classmethod
  def make_loss_fn(