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')
Original: https://www.cnblogs.com/camilia/p/15976148.html
Author: CAMILIA
Title: [报错]-RuntimeError: Input type (torch.cuda.HalfTensor) and weight type (torch.cuda.FloatTensor) should be the same
原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/683361/
转载文章受原作者版权保护。转载请注明原作者出处!