Update is messed up when mmyolo adds new learnable parameters for training.

[lpz7]

When I add the quantization module learning to yolov5, the newly added learnable parameters are not updated, even when I add my own parameters to YOLOv5OptimizerConstructor.

After the following error, I added find_unused_parameters=True and the error disappeared, but the customized parameters still did not participate in the training.

[doonny]

having the same issue on my side ...