No module named 'torch.cuda.amp.autocast'

vision
1

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

2

Answered here.
Please try to avoid double posts. :wink: