original_docstring: 
      A context where the model is temporarily changed to eval mode,
      and restored to previous mode afterwards.
  
      Args:
          model: a torch Module
      
docstring: A context where the model is temporarily changed to eval mode,
  and restored to previous mode afterwards.
docstring_tokens: A context where the model is temporarily changed to eval mode and restored to previous mode afterwards .
code: def inference_context(model):
      """
      A context where the model is temporarily changed to eval mode,
      and restored to previous mode afterwards.
  
      Args:
          model: a torch Module
      """
      training_mode = model.training
      model.eval()
      yield
      model.train(training_mode)
code_tokens: def inference_context ( model ) : training_mode = model . training model . eval ( ) yield model . train ( training_mode )
short_docstring: A context where the model is temporarily changed to eval mode,
  and restored to previous mode afterwards.
short_docstring_tokens: A context where the model is temporarily changed to eval mode and restored to previous mode afterwards .
comment: """
      A context where the model is temporarily changed to eval mode,
      and restored to previous mode afterwards.
  
      Args:
          model: a torch Module
      """