apex amp state dict
🐛 Bug

  
    
      pytorch-lightning/pytorch_lightning/trainer/training_io.py
    
    
         Line 310
      in
      25ee51b
    
  
  
    

        
          
           if self.use_amp and NATIVE_AMP_AVALAIBLE and 'native_amp_scaling_state' in checkpoint: 
        
    
  


It seems for native amp support, the scalar state dict is saved. But for non native amp, the amp state dict is not saved?