使用PyTorch进行模型训练的时候,遇到了错误: RuntimeError: expected scalar type Half but found Float 有一种可能的原因是训练的时候使用了FP16和其他的精度,但是有些显卡不支持混合精度,比如Nvidia的V100显卡,这样就不能使用FP16这个精度进行训练。 在你的模型代码中,将: fp16=True 改为: fp16=False 有可能可以解决问题。
使用PyTorch进行模型训练的时候,遇到了错误: RuntimeError: expected scalar type Half but found Float 有一种可能的原因是训练的时候使用了FP16和其他的精度,但是有些显卡不支持混合精度,比如Nvidia的V100显卡,这样就不能使用FP16这个精度进行训练。 在你的模型代码中,将: fp16=True 改为: fp16=False 有可能可以解决问题。
