Many freshmen just started using pytorch and didn’t know to use model.train() and model.eval().  If you train the model, you can add a model.train() in front. If you test the model, you must add a model.eval().