How do you change the number of classes?

[lm17918]

Given a dataset how do you change the number of classes for the robust label map in the discriminator? there a number: 194 that is used in the disciminator, can you clarify what is that and why is hardcoded instead of coming from a config variable?

[PrasanH]

I think that's because there are 194 classes in the MSeg (which is used to generate the robust labels)