Hi all,
when trying to use the native amp functionalities I run in the following error:
RuntimeError: Input type (torch.cuda.DoubleTensor) and weight type (torch.cuda.HalfTensor) should be the same

I use the @autocast() decorator and the GradScaler class.
PyTorch version is: 1.7.0.dev20200716
Thanks in advance for any hints or help!