Bug in knet_head

[wusize]

Thanks for your error report and we appreciate it a lot.

Checklist

1. I have searched related issues but cannot get the expected help.
2. The bug has not been fixed in the latest version.

Describe the bug
A clear and concise description of what the bug is.

Reproduction

1. What command or script did you run?

   A placeholder for the command.
   

2. Did you make any modifications on the code or config? Did you understand what you have modified?

3. What dataset did you use?

Environment

1. Please run python mmseg/utils/collect_env.py to collect necessary environment information and paste it here.
2. You may add addition that may be helpful for locating the problem, such as
   • How you installed PyTorch [e.g., pip, conda, source]
   • Other environment variables that may be related (such as $PATH, $LD_LIBRARY_PATH, $PYTHONPATH, etc.)

Error traceback

If applicable, paste the error trackback here.

A placeholder for trackback.

Bug fix

In knet_head.py L414.

[wusize]

Add "self.out_channels = self.num_classes" to the init function

[MengzhangLI]

Hi, thanks for your feedback. We would figure out this problem and update to you asap.

Best,