tiny ppl doc typo fix (huggingface#5751)

pnf · Jul 14, 2020 · 5d17895 · 5d17895
1 parent ac921f0
commit 5d17895
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/source/perplexity.rst b/docs/source/perplexity.rst
@@ -123,7 +123,7 @@ are 512 preceding tokens available to condition on).
     stride = 512
 
     lls = []
-    for i in tqdm(range(1, encodings.input_ids.size(1), stride)):
+    for i in tqdm(range(0, encodings.input_ids.size(1), stride)):
         begin_loc = max(i + stride - max_length, 0)
         end_loc = i + stride
         input_ids = encodings.input_ids[:,begin_loc:end_loc].to(device)