您训练好后权重文件的问题

[workerlk]

您好，最近使用了您的权重文件，加载模型时候，说键名不对，后来改好键名之后，说模型的维度不对，请问您是更改过模型吗，还是我哪里出了问题
size mismatch for arcface.weight: copying a param with shape torch.Size([41, 128]) from checkpoint, the shape in current model is torch.Size([1, 128]).
错误类似这样的，如果更改过，您可以分享一下新的权重文件吗

[liuyoude]

权重文件训练时模型的输出类别数量为41（6个machine type的总id数量），该报错是因为当前输出类别数量设置为1，输出类别数量参考该行代码：https://github.com/liuyoude/STgram-MFN/blob/d6c886559ed3d2f0ff3e35b289d391addcedbd7d/run.py#L40C51-L40C51

[Vikllo]

您好，最近使用了您的权重文件，加载模型时候，说键名不对，后来改好键名之后，说模型的维度不对，请问您是更改过模型吗，还是我哪里出了问题 size mismatch for arcface.weight: copying a param with shape torch.Size([41, 128]) from checkpoint, the shape in current model is torch.Size([1, 128]). 错误类似这样的，如果更改过，您可以分享一下新的权重文件吗

您好，我在加载模型的时候也遇到了键名称不正确的问题，请问您是如何修改的呀？另外还想问下您有没有解决模型尺寸不正确的问题，如果已经解决了还想再向您请教一下，望回复。

[workerlk]

谢谢，后边没看见您的回复，我当初第一次使用GitHub，不太熟悉这个怎么使用，现在才想到回复