I am training a model using google colab and i got this error when i am trying to import autocast
code used to import autocast

import torch.cuda.amp.autocast

error i obtained is
ModuleNotFoundError: No module named 'torch.cuda.amp.autocast'

Thank you