TypeError: empty(): argument ‘size‘ must be tuple of ints, but found element of type Tensor at pos 1

在运行代码时,遇到标题问题。
在这里插入图片描述

方法一:检查实例化模型的名称是否正确

m_Xception = Xception()#实例化名称
output = Xception(img.to(device))#模型加载数据

原模型传入参数为标签种类个数,应为整数型
而传入参数是数据Tensor类型,所以报错
改之后的代码

output = m_Xception(img.to(device))

没有再报标题错误

方法二:检查代入参数类型

你可能感兴趣的:(计算机视觉,计算机视觉,人工智能,图像处理)