错误：
RuntimeError: expected scalar type Double but found Float
解决：
pytorch使用gpu加速时，数据类型不支持double，也就是float64,需要将其数据类型变为float.32
【pytorch】深度学习模型踩坑
猜你喜欢
转载自blog.csdn.net/aaatomaaa/article/details/133429218
今日推荐
周排行