get_loss_fun err message

[flystarhe]

pycls/pycls/core/builders.py

Line 35 in 35429e8

assert cfg.MODEL.LOSS_FUN in _loss_funs.keys(), err_str.format(cfg.TRAIN.LOSS)

assert cfg.MODEL.LOSS_FUN in _loss_funs.keys(), err_str.format(cfg.TRAIN.LOSS)

To:

assert cfg.MODEL.LOSS_FUN in _loss_funs.keys(), err_str.format(cfg.MODEL.LOSS)

[flystarhe]

assert cfg.MODEL.LOSS_FUN in _loss_funs.keys(), err_str.format(cfg.MODEL.LOSS_FUN)

[pdollar]

Sorry just getting to this now. Nice catch! Interestingly, this bug has existed since our very first release of pycls.

@flystarhe do you want to create a PR? Otherwise, I'll fix this on my side soon.

Thanks!