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"]
code: def forward(self, *args, use_r3f: bool = False, **kwargs):
      if use_r3f:
          # forward with the normal model
          model_output = self.original_forward(
              *args,
              **kwargs,
          )
  
          # compute noised model outputs
          noise_model_outputs = self.forward_with_noise(
              *args,
              **kwargs,
          )
  
          return model_output, noise_model_outputs
      else:
          return self.original_forward(*args, **kwargs)
  
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 7 for calling the R3FPyTextMixin obj's underlying member method code with example usage: obj.forward() and returns: model_output,  noise_model_outputs, self