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):