Update README.md

Author: VainF

README.md

@@ -5,8 +5,6 @@ Pretrained DeepLabv3, DeepLabv3+ for Pascal VOC & Cityscapes.
 ## Quick Start 
 
 ### 1. Available Architectures
-Specify the model architecture with '--model ARCH_NAME' and set the output stride using '--output_stride OUTPUT_STRIDE'.
-
 | DeepLabV3    |  DeepLabV3+        |
 | :---: | :---:     |
 |deeplabv3_resnet50|deeplabv3plus_resnet50|
@@ -16,12 +14,15 @@ Specify the model architecture with '--model ARCH_NAME' and set the output strid
 |deeplabv3_hrnetv2_32 | deeplabv3plus_hrnetv2_32 |
 |deeplabv3_xception | deeplabv3plus_xception |
 
+please refer to [network/modeling.py](https://github.com/VainF/DeepLabV3Plus-Pytorch/blob/master/network/modeling.py) for all model entries.
+
 All pretrained models: [Dropbox](https://www.dropbox.com/sh/w3z9z8lqpi8b2w7/AAB0vkl4F5vy6HdIhmRCTKHSa?dl=0), [Tencent Weiyun](https://share.weiyun.com/qqx78Pv5)
 
 Note: The HRNet backbone was contributed by @timothylimyl. A pre-trained backbone is available at [google drive](https://drive.google.com/file/d/1NxCK7Zgn5PmeS7W1jYLt5J9E0RRZ2oyF/view?usp=sharing).
 
 ### 2. Load the pretrained model:
 ```python
+model = 
 model.load_state_dict( torch.load( CKPT_PATH )['model_state']  )
 ```
 ### 3. Visualize segmentation outputs: