Pytorch的训练模式和测试模式切换
由于训练的时候Dropout和BN层起作用，每个batch BN层的参数不一样，dropout在训练时随机失效点具有随机性，所以训练和测试要区分开来。
使用时切记要根据实际情况切换：
model.train()model.eval()
切记！！！！！！！！
究竟错几遍才能记住呢？让我们拭目以待！ -.-
posted on 2019-04-09 22:45 那抹阳光1994 阅读(1032) 评论(0) 编辑 收藏 举报
