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

model.train() :启用 BatchNormalization 和 Dropout

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

你可能感兴趣的:(Pytorch中的 model.train() 和 model.eval() 模式)