RuntimeError: mat1 and mat2 must have the same dtype

一个快速的解决方法可能是替换dtype='float32’的浮点数。np默认返回float64类型。F.linear对精度傻了。不如去stackoverflow上看一眼。

likeGhee

20079人浏览 · 2022-11-12 20:19:16

likeGhee · 2022-11-12 20:19:16 发布

这个坑卡了我一晚上

不如去stackoverflow上看一眼

np默认返回float64类型。F.linear对精度傻了。

一个快速的解决方法可能是换成dtype='float32’的浮点数

inputs = inputs.to(torch.float32)

2048 AI社区

有“AI”的1024 = 2048，欢迎大家加入2048 AI社区

更多推荐

从「能用」到「可靠」：深入探讨C++异常安全

目标明确：首先追求基本保证（无泄漏），这是底线。然后，对于关键操作，努力实现强保证。拥抱 RAII：这是你最重要的工具。用智能指针、容器管理资源，对于自定义资源，封装成 RAII 类。善用 "Copy-and-Swap"：这是实现强保证函数的一个通用且有效的方法。正确使用noexcept：为移动操作、swap和析构函数标记noexcept。严守铁律：决不让异常从析构函数中逃逸。异常安全不是事后添加