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