Pytorch中的 model.train() 和 model.eval() 模式

model.train() :启用 BatchNormalization 和 Dropout

model.eval() :不启用 BatchNormalization 和 Dropout

 把模型model.eval()作用是为了固定BN和dropout层,使得偏置参数不随着发生变化。因为当batchsize小时,如果没有固定,会对图像的失真有很大的影响。

参考链接:https://pytorch.org/docs/stable/nn.html?highlight=module%20eval#torch.nn.Module.eval

posted on 2020-08-21 22:58  cltt  阅读(390)  评论(0编辑  收藏  举报

导航