From 19c6cbaa979e849de5bec9980673fab3fec541a1 Mon Sep 17 00:00:00 2001
From: Kosi Asuzu <34872572+asuzukosi@users.noreply.github.com>
Date: Wed, 26 Jun 2024 11:33:47 +0100
Subject: [PATCH] fix typo

correct `model.train` description to be 'Put model into training mode' as opposed to 'Put model into inference mode'
---
 language_translation/main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/language_translation/main.py b/language_translation/main.py
index 2b4fbb94c3..cc0d199cd8 100644
--- a/language_translation/main.py
+++ b/language_translation/main.py
@@ -111,7 +111,7 @@ def train(model, train_dl, loss_fn, optim, special_symbols, opts):
     # Object for accumulating losses
     losses = 0
 
-    # Put model into inference mode
+    # Put model into training mode
     model.train()
     for src, tgt in tqdm(train_dl, ascii=True):