- Fix to read `import torch.nn.function as F` and - Include `model.train()` call in the lesson text
import torch.nn.function as Fandmodel.train()call in the lesson text