Shulou(Shulou.com)06/02 Report--

What is the function of the train () method in PyTorch? I believe many inexperienced people don't know what to do about it. Therefore, this paper summarizes the causes and solutions of the problem. Through this article, I hope you can solve this problem.

Pytorch provides us with two ways to switch between training and evaluation (inference) modes. They are: model.train () and model.eval () dropout work during training, being bypassed during inference does not work, or equivalently, the probability is set to zero. Batch normalization, like dropout, has two modes, corresponding to training and inference, respectively: model.eval () and model.train ().

The following is a screenshot of the original text:

Dropout:

Batch normalization:

Official API screenshot of PyTorch:

Train (mode=True):

Eval ():

After reading the above, have you mastered the function of the train () method in PyTorch? If you want to learn more skills or want to know more about it, you are welcome to follow the industry information channel, thank you for reading!

Welcome to subscribe "Shulou Technology Information " to get latest news, interesting things and hot topics in the IT industry, and controls the hottest and latest Internet news, technology news and IT industry trends.