Deeplab V3+ MobileNetV2 training error about segmentron HOT 3 OPEN

I can not reproduce your error by using configs/cityscapes_deeplabv3_plus_mobilenet.yaml

from segmentron.

Hello Sir,

I tested your code using my datasets (image resolution : 256).

I was succeed deeplabv3_plus based on mobilenet_v2.
but I was failed deeplabv3_plus based on hrnet_w18_small_v1.

What can I do check file this problems??

Thanks.

from segmentron.

I meet the same problem, this is due the process:
https://github.com/LikeLy-Journey/SegmenTron/blob/master/tools/train.py

if get_rank() == 0:
    try:
        show_flops_params(self.model, args.device)
    except Exception as e:
        logging.warning('get flops and params error: {}'.format(e))

function show_flops_params add some hooks which lead to the error.
Just comment it out.

from segmentron.