改完还是一样的报错
您好,这个错误是因为torch.full_like(x[i], 0)的输入tensor需要是cpu类型的,而这里的x[i]是一个mlu上的tensor,所以报了"RuntimeError: Can not call cpu_data on an empty tensor"这个错误
看起来您这个应该是来自于下面的repodeepcam-cn/yolov5-face: YOLO5Face: Why Reinventing a Face Detector (https://arxiv.org/abs/2105.12931) (github.com)这个地方是属于后处理的部分,不知道您有没有尝试过将后处理部分替换为我们的大算子实现,原始yolov5的后处理我们是有部分算子不支持的
看起来您这个应该是来自于下面的repo
deepcam-cn/yolov5-face: YOLO5Face: Why Reinventing a Face Detector (https://arxiv.org/abs/2105.12931) (github.com)
这个地方是属于后处理的部分,不知道您有没有尝试过将后处理部分替换为我们的大算子实现,原始yolov5的后处理我们是有部分算子不支持的
这个报错看起来像是x[i]是一个空的tensor,可以检查一下x[i]这个tensor的shape和值
shape有打印啊
full_like这里是生成了一个在cpu上的tensor,需要使用to(ct.mlu_device())将其转到mlu上才能正常使用
请登录后评论