你好，想请教一下，用fast-reid提取特征时，怎么将维度设置为512维

问题遇到的现象和发生背景

我想修改fast-reid/demo，提取行人的512维特征，修改前能提取2048维特征

问题相关代码，请勿粘贴截图

修改yml文件的FEAT_DIM为512

运行结果及报错内容

我的解答思路和尝试过的方法

我想要达到的结果

正常运行，得出输入图片的512维特征