Update README.md

VioletRaven · Sep 8, 2022 · 27520c5 · 27520c5
1 parent 00c1b23
commit 27520c5
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -34,13 +34,12 @@ The script will create tokenized .json files that can be fed to the "train.py" s
 
 3. Training
 
-In order to run the training process on a GPU follow the Google Colab Notebook provided as
+In order to run the training process on a GPU follow the Google Colab Notebook provided:
 
-bash_train_GPT2.ipynb
+**bash_train_GPT2.ipynb**
 
 Remember to change the Runtime to GPU !
 
-
 4. Loading and using the model
 
 Once the training has stopped and the best model has been saved you can load it and use it for summarization by running the script below on your terminal.