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

model.train() :启用 BatchNormalization 和 Dropout

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

猜你喜欢

转载自blog.csdn.net/weixin_42535423/article/details/107942715
今日推荐