Pytorch中的 model.train() 和 model.eval() 模式
Posted sbj123456789
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了Pytorch中的 model.train() 和 model.eval() 模式相关的知识，希望对你有一定的参考价值。
model.train() ：启用 BatchNormalization 和 Dropout
model.eval() ：不启用 BatchNormalization 和 Dropout
参考：
https://pytorch.org/docs/stable/nn.html?highlight=module%20eval#torch.nn.Module.eval
以上是关于Pytorch中的 model.train() 和 model.eval() 模式的主要内容，如果未能解决你的问题，请参考以下文章
model.train()model.eval()optimizer.zero_grad()loss.backward()optimizer.step作用及原理详解Pytorch入门手册
model.train()model.eval()optimizer.zero_grad()loss.backward()optimizer.step作用及原理详解Pytorch入门手册
[Pytorch系列-38]： 工具集 - torchvision预定义模型的两种模式model.train和model.eval的表面和本质区别