From dcedf26dd11a2f83bf99029cd2dc8fb31e2cfc66 Mon Sep 17 00:00:00 2001
From: Felix Divo <4403130+felixdivo@users.noreply.github.com>
Date: Tue, 7 Mar 2023 15:08:54 +0100
Subject: [PATCH] Fix number of iterations in example code

---
 beginner_source/flava_finetuning_tutorial.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/beginner_source/flava_finetuning_tutorial.py b/beginner_source/flava_finetuning_tutorial.py
index 92bc5031709..ef54c5ea06d 100644
--- a/beginner_source/flava_finetuning_tutorial.py
+++ b/beginner_source/flava_finetuning_tutorial.py
@@ -172,7 +172,7 @@ def transform(tokenizer, input):
     loss.backward()
     optimizer.step()
     print(f"Loss at step {idx} = {loss}")
-    if idx > MAX_STEPS-1:
+    if idx >= MAX_STEPS-1:
       break