[New Models] 自己实现的检测头权重在评估过程中无法正确加载

[UncleNiNi]

Model/Dataset/Scheduler description

我自己实现了一个自定义的bbox头，在这个实现bbox头的文件里我自定义了一个模块。
使用了我的自定义bbox头的模型训练过程是正常的，但是在评估过程中，我发现模型表现不正常，经过排查之后发现是因为我在bbox头文件里自定义模块的训练好的权重在评估中没有被加载，而是随机初始化的。请问我该如何解决这个问题？

Open source status

• The model implementation is available
• The model weights are available.

Provide useful links for the implementation

No response