摘要:
一、出错误背景: Pytorch 中想使用 CUDA 对程序计算进行加速 二、问题分析 错误的意思:object 的 device 类型期望得到的是 cuda 类型,但是实际上的类型确实 cpu 类型,在调用二分类交叉熵损失进行前向计算的时候 三、检查下面几点: 模型是否放到了CUDA上 model 阅读全文
posted @ 2020-12-24 15:55
技术研究与问题解决
阅读(413)
评论(0)
推荐(0)
摘要:
torch.Tensor的4种乘法 阅读全文
posted @ 2020-12-24 13:19
技术研究与问题解决
阅读(348)
评论(0)
推荐(0)
浙公网安备 33010602011771号