fix bugs from issue

configs/deeplabv3p_mobilenetv2_cityscapes.yaml

@@ -30,6 +30,8 @@ MODEL:
         BACKBONE: "mobilenetv2"
         ASPP_WITH_SEP_CONV: True
         DECODER_USE_SEP_CONV: True
+        ENCODER_WITH_ASPP: False
+        ENABLE_DECODER: False
 TRAIN:
     PRETRAINED_MODEL_DIR: u"pretrained_model/deeplabv3p_mobilenetv2-1-0_bn_coco"
     MODEL_SAVE_DIR: "saved_model/deeplabv3p_mobilenetv2_cityscapes"

pdseg/models/model_builder.py

@@ -85,7 +85,7 @@ def seg_model(image, class_num):
         logits = fast_scnn.fast_scnn(image, class_num)
     else:
         raise Exception(
-            "unknow model name, only support unet, deeplabv3p, icnet, pspnet, hrnet"
+            "unknow model name, only support unet, deeplabv3p, icnet, pspnet, hrnet, fast_scnn"
         )
     return logits