2 ms·

Thanks, it's great to know that I can use PyTorch AMP for mixed precision training, I was following NVIDIA's documentation, and this wasn't explained.