https://python.elitedev.in/deep_learning/mixed-precision-training-pytorch-faster-deep-learning/