我用Trained Models and logs (Size 256) 下面表格中的模型进行评估测试，报错KeyError: 'base.cls_token'

[lilinebo]

请问这是什么原因呢。我看了表格中模型对应的log，模型名字和下载的不一样，是否是这个原因。我看了log里面的配置文件信息，和我加载yml一样，除了模型名字。

[ming132]

我也遇到了这个问题，veri的TransReID用下载的模型跑出来效果也很差，不知道什么情况

[KingH12138]

First, you need to consider the prefix of param-key;
Second, make sure you use the right checkpoint because there are two checkpoint param in .yml;
Last, I guess you had ignored a param in .yml callied "STRIDE_SIZE: [16, 16]". Accordingly, It should be [12, 12] when you apply author's checkpoints. If you deal with point 1,2 this my comment, the mismatch of pos_embedding's size will bother you and you can deal with it by change param STRIDE_SIZE in .yml.

Tips: do not forget to change code of dataset class if you need to have a new logic to load your msmt17 and other dataset.