class balancing option is not accessible
priancho opened this issue · comments
Description
In train.py, MixSoftmaxCrossEntropyOHEMLoss() method (https://github.com/Tramac/Fast-SCNN-pytorch/blob/master/train.py#L103) doesn't allow to control use_weight option, which is True by default.
I think that it should be False by default because this class balancing weights should be applied only to cityscapes dataset.
It might be better to provide class balancing weights from cmd-args or from data_loader class (e.g., data_loader/cityscapes.py).
Yeah, thanks for your advice.