Update stream.rst

huggingface · Dec 3, 2021 · 10ec8c6 · 10ec8c6 · github-actions · Dec 3, 2021
1 parent 520e9ff
commit 10ec8c6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/source/stream.rst b/docs/source/stream.rst
@@ -143,8 +143,8 @@ The following example demonstrates how to tokenize a :class:`datasets.IterableDa
    'attention_mask': [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, ..., 1, 1]}
 
 
-In a training loop
-^^^^^^^^^^^^^^^^^^
+Stream in a training loop
+^^^^^^^^^^^^^^^^^^^^^^^^^
 
 :class:`datasets.IterableDataset`s can be integrated into a training loop. First, shuffle the dataset: