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