06 2022 档案

解决训练yolor时，出现错误：‘Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!’
摘要：问题 ‘Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!’ 解决 方法1：x.to(device) 把 device 作为一个可变参数，推荐使用argparse 阅读全文
posted @ 2022-06-13 15:08 poowicat 阅读(4663) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示