Save scaler state dict when checkpointing
# What does this PR do?

One last thing was missing for resuming with checkpoints and have exactly the same results as a complete training: the gradient scaler state when using mixed precision with AMP in PyTorch. This PR addresses that.

Fixes #11323