2022年6月13日
解决训练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 阅读(3841) 评论(0) 推荐(0) 编辑