代码收藏家 技术教程 输入类型(torch.FloatTensor)和权重类型(torch.cuda.FloatTensor)应保持一致 这个错误提示表明输入数据的类型和模型权重的类型不匹配,可能是因为模型权重是在GPU上...
代码收藏家 技术教程 使用Anomalib项目的padim无监督算法进行自制工业缺陷数据集的模型训练和ONNX部署(二):Python代码解读 上一篇博客中完成了Anomalib中padim算法的模型训练,得到了onnx模型以及...