How to Add Custom Parameters to loss, predict, and _forward Functions in New Segmentor in mmseg1.2.2

[jinqin-99]

Hello, I am using mmsegmentation to develop a new segmentation model. During the process, I found that I need to pass additional parameters through the loss, predict, and _forward functions. These parameters are essential for my custom logic, but the current implementation does not support such inputs.

add new parameters like this:

class NewSegmentor(EncoderDecoder):
    def loss(self, inputs, data_samples, new_param=None):
        pass

    def predict(self, inputs,data_samples, new_param=None):
        pass

    def _forward(self, inputs, data_samples, new_param=None):
        pass

What I should do? I look forward to your help！