Skip to content

Commit

Permalink
fix typo (#1272)
Browse files Browse the repository at this point in the history
correct `model.train` description to be 'Put model into training mode' as opposed to 'Put model into inference mode'
  • Loading branch information
asuzukosi authored Feb 9, 2025
1 parent 1bef748 commit 5dfeb46
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion language_translation/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):

Expand Down

0 comments on commit 5dfeb46

Please sign in to comment.