title: ImportError: cannot import name 'SAVE_STATE_WARNING' from 'torch.optim.lr_scheduler'
body: <p>I am attempting to issue this statement in a jupyter Notebook.</p>
  <pre><code>from transformers import BertForQuestionAnswering
  </code></pre>
  <p>I get the error:</p>
  <blockquote>
  <p>ImportError: cannot import name 'SAVE_STATE_WARNING' from 'torch.optim.lr_scheduler' (C:\Users\sbing.conda\envs\Tensorflow\lib\site-packages\torch\optim\lr_scheduler.py)</p>
  </blockquote>
  <p>Here is the complete stack:</p>
  <blockquote>
  <p>ImportError                               Traceback (most recent call last)
   in 
  ----&gt; 1 from transformers import BertForQuestionAnswering</p>
  <p>~.conda\envs\Tensorflow\lib\site-packages\transformers_<em>init</em>_.py in 
  624
  625     # Trainer
  --&gt; 626     from .trainer import Trainer
  627     from .trainer_pt_utils import torch_distributed_zero_first
  628 else:</p>
  <p>~.conda\envs\Tensorflow\lib\site-packages\transformers\trainer.py in 
  67     TrainerState,
  68 )
  ---&gt; 69 from .trainer_pt_utils import (
  70     DistributedTensorGatherer,
  71     SequentialDistributedSampler,</p>
  <p>~.conda\envs\Tensorflow\lib\site-packages\transformers\trainer_pt_utils.py in 
  38     SAVE_STATE_WARNING = &quot;&quot;
  39 else:
  ---&gt; 40     from torch.optim.lr_scheduler import SAVE_STATE_WARNING
  41
  42 logger = logging.get_logger(<strong>name</strong>)</p>
  <p>ImportError: cannot import name 'SAVE_STATE_WARNING' from 'torch.optim.lr_scheduler' (C:\Users\sbing.conda\envs\Tensorflow\lib\site-packages\torch\optim\lr_scheduler.py)</p>
  </blockquote>