【pytorch RuntimeError: expected scalar type Double but found Float】深度学习模型踩坑

错误:
RuntimeError: expected scalar type Double but found Float
解决:
pytorch使用gpu加速时,数据类型不支持double,也就是float64,需要将其数据类型变为float.32

你可能感兴趣的:(深度学习,pytorch,人工智能)