Skip to content

Commit 5dfeb46

Browse files
authored
fix typo (#1272)
correct `model.train` description to be 'Put model into training mode' as opposed to 'Put model into inference mode'
1 parent 1bef748 commit 5dfeb46

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

language_translation/main.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,7 @@ def train(model, train_dl, loss_fn, optim, special_symbols, opts):
111111
# Object for accumulating losses
112112
losses = 0
113113

114-
# Put model into inference mode
114+
# Put model into training mode
115115
model.train()
116116
for src, tgt in tqdm(train_dl, ascii=True):
117117

0 commit comments

Comments
 (0)