[報錯]-RuntimeError: Input type (torch.cuda.HalfTensor) and weight type (torch.cuda.FloatTensor) should be the same

RuntimeError: Input type (torch.cuda.HalfTensor) and weight type (torch.cuda.FloatTensor) should be the same

---

模型輸入的數據類型要與模型參數的數據類型一致。
torch.cuda.HalfTensor：對應

 np.array(x, dtype = 'float32')

torch.cuda.FloatTensor：對應

 np.array(x, dtype = 'float16')

參考鏈接：
https://stackoverflow.com/questions/65029217/runtimeerror-input-type-torch-cuda-floattensor-and-weight-type-torch-cuda-ha