2024-8-31
发现测试代码用不了的原因是因为pkl文件生成的有问题!
用这个代码生成了距离和树形数据就可以使用训练好的模型进行test了、
预测生成的每个结果是class包含的,只能是class包含的(目前来说)
------------------------------------------跑一下demo----------------------------------------------------------------------
报错 Expected size for first two dimensions of batch2 tensor to be: [8, 1000] but got: [8, 23].
解决方案:没有选择swint 而是用的resnet 张量不匹配
demo的评估结果如下: