class_docstr: Represents a RL policy + value function, containing a value and logit head.
  Only returns pre-token values (never evaluates on the final response token)
code: def __init__(
      self,
      task_hparams: tasks.TaskHParams = None,
      logit_head=True,
      **kwargs,
  ):
      super().__init__(
          logit_head=logit_head, heads=("value",), **kwargs
      )
      self.task_hparams = task_hparams
  
file_path: summarize-from-feedback/summarize_from_feedback/policy.py
imports: import functools
  from summarize_from_feedback import tasks
  from summarize_from_feedback.query_response_model import QueryResponseModel
label_desc: Use image node_id 1 to create a new Policy object from inherited base classes: QueryResponseModel with example: obj = Policy(task_hparams, logit_head)