在运行代码时,遇到标题问题。
方法一:检查实例化模型的名称是否正确
m_Xception = Xception()#实例化名称
output = Xception(img.to(device))#模型加载数据
原模型传入参数为标签种类个数,应为整数型
而传入参数是数据Tensor类型,所以报错
改之后的代码
output = m_Xception(img.to(device))
没有再报标题错误
方法二:检查代入参数类型
在运行代码时,遇到标题问题。
方法一:检查实例化模型的名称是否正确
m_Xception = Xception()#实例化名称
output = Xception(img.to(device))#模型加载数据
原模型传入参数为标签种类个数,应为整数型
而传入参数是数据Tensor类型,所以报错
改之后的代码
output = m_Xception(img.to(device))
没有再报标题错误
方法二:检查代入参数类型