ModuleNotFoundError: No module named 'mmseg.models.backbones.pvt_dia'

[Hzhang818]

Hello, after I followed the installation instruction and tried to import init_segmentor, it ran into the following problem.
/usr/local/python-3.6.8/lib/python3.6/site-packages/mmsegmentation-0.11.0-py3.6.egg/mmseg/models/backbones/init.py in
9 from .unet import UNet
10 from .pvt import pvt_small, pvt_small_f4, pvt_tiny
---> 11 from .pvt_dia import pvt_small_dia
12 from .vit import VisionTransformer
13 from .vit_mla import VIT_MLA

ModuleNotFoundError: No module named 'mmseg.models.backbones.pvt_dia'

Would you mind telling me how should I solve the problem. Thanks in advance.

[XiongweiWu]

@Hzhang818 Oh it's a variant of PVT model I use in my experiments which is not used finally. You can just delete this line of code and I have also fixed this bug. Thx for your issue!