解决RuntimeError: expected scalar type Half but found Float问题

把训练或者推理部分实际提交到GPU的代码前增加下面这句并缩进。 `with torch.cuda.amp.autocast():`