class_docstr: Mixin class for applying the R3F method, to apply R3F with any model
  inherit the class and implement the abstract functions.
  
  For more details: https://arxiv.org/abs/2008.03156
class_docstr_tok: ["Mixin","class","for","applying","the","R3F","method",",","to","apply","R3F","with","any","model","inherit","the","class","and","implement","the","abstract","functions",".","For","more","details",":","https",":","\/\/arxiv.org\/abs\/2008.03156"]
docstr: Extracts the output from the model.forward() call that is used for the
  r3f loss term
docstr_tok: ["Extracts","the","output","from","the","model.forward","(",")","call","that","is","used","for","the","r3f","loss","term"]
code: def get_r3f_model_output(self, model_output):
      """
      Extracts the output from the model.forward() call that is used for the
      r3f loss term
      """
      return model_output
  
imports: from contextlib import AbstractContextManager
  from enum import Enum
  from typing import Dict
  import torch
  import torch.nn.functional
  from pytext.common.constants import Stage
  from pytext.config import ConfigBase
  from pytext.utils.precision import maybe_float
label_desc: Use image node_id 6 for calling the R3FPyTextMixin obj's underlying member method code with example usage: obj.get_r3f_model_output(model_output) and returns: model_output