Update README.md

README.md

@@ -82,7 +82,7 @@ Quick overview of the datasets we provide
 
 All functionality follows the pattern `./deepy.py main_function.py -d configs small.yml local_configs.yml`
 We currently offer four main functions:
-1. `train.py` is used for training and finetuning models
+1. `pretrain_gpt2.py` is used for training and finetuning models
 2. `evaluate.py` is used to evaluate a trained model using the evaluation harness
 3. `generate.py` is used to sample text from a model.
 4. `distill.py` is used to distill a larger model into a smaller model.