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

 @halomaster 06 Nov 2022, 15:30 GMT【AI】 

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

0 replies