错误:
RuntimeError: expected scalar type Double but found Float
解决:
pytorch使用gpu加速时,数据类型不支持double,也就是float64,需要将其数据类型变为float.32
【pytorch RuntimeError: expected scalar type Double but found Float】深度学习模型踩坑
于 2023-09-30 10:50:42 首次发布
错误:
RuntimeError: expected scalar type Double but found Float
解决:
pytorch使用gpu加速时,数据类型不支持double,也就是float64,需要将其数据类型变为float.32