Pytorch的训练模式和测试模式切换
由于训练的时候Dropout和BN层起作用，每个batch BN层的参数不一样，dropout在训练时随机失效点具有随机性，所以训练和测试要区分开来。
使用时切记要根据实际情况切换：
model.train()
model.eval()
切记！！！！！！！！
究竟错几遍才能记住呢？让我们拭目以待！ -.-
相关文章：
相似解决方案
热门标签
Pytorch的训练模式和测试模式切换
由于训练的时候Dropout和BN层起作用，每个batch BN层的参数不一样，dropout在训练时随机失效点具有随机性，所以训练和测试要区分开来。
使用时切记要根据实际情况切换：
model.train()
model.eval()
切记！！！！！！！！
究竟错几遍才能记住呢？让我们拭目以待！ -.-
相关文章：