Dropout和BN层的模式切换

Pytorch的训练模式和测试模式切换

由于训练的时候Dropout和BN层起作用，每个batch BN层的参数不一样，dropout在训练时随机失效点具有随机性，所以训练和测试要区分开来。

使用时切记要根据实际情况切换：

model.train()
model.eval()

切记！！！！！！！！

究竟错几遍才能记住呢？让我们拭目以待！    -.-

【推广】 免费学中医，健康全家人
原文地址：https://www.cnblogs.com/jiangkejie/p/10680403.html