Method load_weights

language/mentionmemory/tasks/base_task.py:168–177

Load model weights from file. Args: config: experiment config. Returns: Dictionary of model weights.

(cls, config: ml_collections.ConfigDict)

  @classmethod
  def load_weights(cls, config: ml_collections.ConfigDict) -> Dict[str, Any]:
    """Load model weights from file.

    Args:
      config: experiment config.

    Returns:
      Dictionary of model weights.
    """
    raise NotImplementedError

  @classmethod
  def make_output_postprocess_fn(