code: def on_loss_begin(self, last_output:Tuple[Tensor,Tensor,Tensor], **kwargs):
          "Save the extra outputs for later and only returns the true output."
          self.raw_out,self.out = last_output[1],last_output[2]
          return {'last_output': last_output[0]f on_loss_begin(self, last_output:Tuple[Tensor,Tensor,Tensor], **kwargs):
          "Save the extra outputs for later and only returns the true output."
          self.raw_out,self.out = last_output[1],last_output[2]
          return {'last_output': last_output[0]}
docs: Save the extra outputs for later and only returns the true output.