update read me

Author: wbw520

README.md

@@ -25,8 +25,8 @@ Use the following command for the inference of a sample. You can change the inde
 python vis_recon.py --num_classes 10 --num_cpt 20 --index 0 --top_sample 10 ---deactivate -1
 ```
 
-#### Usage for CUB200, ImageNet, Synthetic and Custom
-We first pre-train the backbone and then train the whole model. For ImageNet, Synthetic and Custom, just change the name for dataset.
+#### Usage for CUB200, ImageNet, Synthetic (matplot) and Custom
+We first pre-train the backbone and then train the whole model. For ImageNet, Synthetic (matplot) and Custom, just change the name for dataset.
 ```
 Pre-training of backbone:
 python main_retri.py --num_classes 50 --num_cpt 20 --base_model resnet18 --lr 0.0005 --epoch 60 --lr_drop 40 --pre_train True --dataset CUB200 --dataset_dir "your dir"