模型训练错误RuntimeError: expected scalar type Half but found Float

点击查看原文

原文作者：老杨 | 来源：老杨说话的地方

使用PyTorch进行模型训练的时候，遇到了错误：

RuntimeError: expected scalar type Half but found Float

有一种可能的原因是训练的时候使用了FP1