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

文章来自:https://www.cnblogs.com/sbj123456789/p/10825685.html
© 2021 jiaocheng.bubufx.com  联系我们
ICP备案:鲁ICP备09046678号-3