I just got the following error when attempting to use amp. Do you know how I can fix it?
     84         if amp_enable:
---> 85             with th.cuda.amp.autocast():
     86                 out1 = model(sub, inp)
     87                 out2 = temp_ly(sub, out1)

AttributeError: module 'torch.cuda.amp' has no attribute 'autocast'