Skip to content

Commit

Permalink
fix typo
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 Jun 26, 2024
1 parent a38cbfc commit 19c6cba
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 19c6cba

Please sign in to comment.