torch.from_numpy(target) TypeError: can't convert np.ndarray of type numpy.object_. 怎么解决
时间: 2023-09-26 21:04:50 浏览: 153
将numpy.ndarray格式图像转化为_io.BufferedReader格式
这个错误通常是因为“target”数组包含非数字或非数值类型的元素。在转换为张量之前,您需要确保数组中所有元素都是数字类型。您可以尝试使用Numpy中的astype()函数将整个数组的类型转换为数字类型,例如:
target = target.astype(np.float32)
tensor_target = torch.from_numpy(target)
如果这个方法没有解决问题,请检查“target”数组中是否存在非数字元素,并尝试将它们从数组中删除。
阅读全文