Update README.md

README.md

@@ -13,7 +13,7 @@ please cite the paper as:
 }
 ```
 
-For example, the file in `gpt2-medium/wikipedia/6_9.pickle` contains the matrix trained to transform 6th-layer hidden representations of tokens into 9th-layer hidden representations, for the Huggingface transformers `gpt2-medium` model. One loads and multiplies as follows:
+For example, the file in `gpt2-medium/wikipedia/6_9.pickle` contains the matrix trained, on the wikipedia dataset, to transform 6th-layer hidden representations of tokens into 9th-layer hidden representations, for the Huggingface transformers `gpt2-medium` model. One loads and multiplies as follows:
 
 ```
 import pickle