model.train() :启用 BatchNormalization 和 Dropout

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

SRE实战 互联网时代守护先锋,助力企业售后服务体系运筹帷幄!一键直达领取阿里云限量特价优惠。

 

参考:

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

扫码关注我们
微信号:SRE实战
拒绝背锅 运筹帷幄