KeyError: 'early_stop_callback_wait' when calling training with auto_lr_find
🐛 Bug

Set auto_lr_find = True and try to train. Training with false flag works well.
To Reproduce
Steps to reproduce the behavior:
model = LightningCanserClassifier(config)
trainer = pl.Trainer(logger=tb_logger, early_stop_callback=early_stop_callback,
                             checkpoint_callback=checkpoint_callback, **config['training']['Trainer'])
trainer.fit(model, train_dataloader=train_loader, val_dataloaders=valid_loader)


 /home/dex/anaconda3/bin/python "/home/dex/Desktop/panda canser kaggle/train.py"
INFO:lightning:GPU available: True, used: True
INFO:lightning:CUDA_VISIBLE_DEVICES: [0]
Finding best initial lr: 100%|██████████| 100/100 [00:35<00:00,  2.72it/s]Traceback (most recent call last):
  File "/home/dex/Desktop/panda canser kaggle/train.py", line 144, in <module>
    trainer.fit(model, train_dataloader=train_loader, val_dataloaders=valid_loader)
  File "/home/dex/anaconda3/lib/python3.7/site-packages/pytorch_lightning/trainer/trainer.py", line 734, in fit
    self._run_lr_finder_internally(model)
  File "/home/dex/anaconda3/lib/python3.7/site-packages/pytorch_lightning/trainer/lr_finder.py", line 31, in _run_lr_finder_internally
    lr_finder = self.lr_find(model)
  File "/home/dex/anaconda3/lib/python3.7/site-packages/pytorch_lightning/trainer/lr_finder.py", line 164, in lr_find
    self.restore(str(save_path), on_gpu=self.on_gpu)
  File "/home/dex/anaconda3/lib/python3.7/site-packages/pytorch_lightning/trainer/training_io.py", line 289, in restore
    self.restore_training_state(checkpoint)
  File "/home/dex/anaconda3/lib/python3.7/site-packages/pytorch_lightning/trainer/training_io.py", line 372, in restore_training_state
    self.early_stop_callback.wait = checkpoint['early_stop_callback_wait']
KeyError: 'early_stop_callback_wait'

Expected behavior

Environment
Please copy and paste the output from our
environment collection script
(or fill out the checklist below manually).
You can get the script and run it with:
wget https://raw.githubusercontent.com/PyTorchLightning/pytorch-lightning/master/tests/collect_env_details.py
# For security purposes, please check the contents of collect_env.py before running it.
python collect_env.py


PyTorch Version (e.g., 1.0): 1.3.1
OS (e.g., Linux): ubuntu
How you installed PyTorch (conda, pip, source): conda
Build command you used (if compiling from source):
Python version: 3.7
CUDA/cuDNN version: cuda10.0.130_cudnn7.6.3_0
GPU models and configuration: 1070ti
Any other relevant information: