Update README.md

Author: Vítor Albiero

README.md

@@ -19,17 +19,17 @@ If you want to retrain on your own dataset, aligned the images first and create
 To train the attribute predictors, you will need to pass the path to images main folder, along with the image list, or an image list that contains the absolute path to the images.
 
 ```
-python3 main.py --train_source /path_to_train_dataset_main_folder/ --train_list ./datasets/age_train.txt --val_source ../path_to_val_dataset_main_folder/ --val_list ./datasets/age_val.tx -a age --prefix age --multi_gpu
+python3 train.py --train_source /path_to_train_dataset_main_folder/ --train_list ./datasets/age_train.txt --val_source ../path_to_val_dataset_main_folder/ --val_list ./datasets/age_val.tx -a age --prefix age --multi_gpu
 ```
 An alternate faster way to train is to convert the datasets to LMDB format. For this end, use the [imagelist2lmdb.py](https://github.com/vitoralbiero/face_analysis_pytorch/blob/master/utils/imagelist2lmdb.py) or [folder2lmdb.py](https://github.com/vitoralbiero/face_analysis_pytorch/blob/master/utils/folder2lmdb.py) to convert a dataset to LMDB. Then, train using the command below.
 ```
-python3 main.py --train_source ./train_dataset.lmdb --val_source ./val_dataset.lmdb/ --val_list ./datasets/age_val.tx -a age --prefix age --multi_gpu
+python3 train.py --train_source ./train_dataset.lmdb --val_source ./val_dataset.lmdb/ --val_list ./datasets/age_val.tx -a age --prefix age --multi_gpu
 ```
 ### Recognition
 To train for recognition, the [LFW, CFP-FP and AgeDB-30](https://github.com/deepinsight/insightface) should be converted using [utils/prepare_test_sets.py](https://github.com/vitoralbiero/face_analysis_pytorch/blob/master/utils/prepare_test_sets.py).
 
 ```
-python3 main.py --train_source ./ms1m_v2.lmdb --val_source ./path_to_val_datasets/ --val_list ['lfw', 'cpf_fp', 'agedb_30'] -a recognition --prefix arcface --multi_gpu --head arcface
+python3 train.py --train_source ./ms1m_v2.lmdb --val_source ./path_to_val_datasets/ --val_list ['lfw', 'cpf_fp', 'agedb_30'] -a recognition --prefix arcface --multi_gpu --head arcface
 ```
 
 If you train using [ArcFace](https://arxiv.org/abs/1801.07698) or [CosFace](https://arxiv.org/abs/1801.09414), please cite the apppropriate papers.