AI article

model.train() vs model.eval() vs torch.no_grad() in PyTorch: What Each One Changes

Community description: PyTorch has three calls that are used around evaluation: model.train(), model.eval() and...

Dev.to | Oct 10, 2026 | Wesam Khallaf — Author of PyTorch From Ground Up

Automated excerpt

A new model starts in training mode. model. eval() is the same call as model. train(False). In evaluation mode it returns its input unchanged. tensor([0. , 0. , 2. , 0. , 0. , 0. , 2. , 2. , 0. , 2. ]) tensor([1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. ]) With p = 0. 5, the surviving elements are multiplied by 2. Validating in training mode also changes the model.

Selected automatically from source text; not independently written or fact-checked. Read the original for full context.

Read the original article

Read next

AI briefing: recent picks

More stories to explore