Old code, why not use PyTorch 1.7.1 with mixed precision

[tadam98]

Hi,

The code used in https://learnopencv.com/how-to-convert-a-model-from-pytorch-to-tensorrt-and-speed-up-inference/ is very old. wht not use mixed precision offered automatically with PyTorch 1.7.1 (with its corresponding torchvision 0.82) autocast ? and then compare ?

Best, Mickey