其他分享
首页 > 其他分享> > Pytorch中的 model.train() 和 model.eval() 模式

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

标签:nn,BatchNormalization,Dropout,Pytorch,train,eval,model
来源: https://www.cnblogs.com/tingtin/p/13543785.html