Hey, I’ve been having trouble to see what’s wrong here. I have torch 1.8.1+cu111
I believe the torch.amp namespace was added in PyTorch 1.12.0+ after mixed-precision training was implemented for the CPU.
In older versions, you would need to use torch.cuda.amp instead.
Thank you! The issue was resolved