Dropout和BN层的模式切换
摘要:
Pytorch的训练模式和测试模式切换 由于训练的时候Dropout和BN层起作用,每个batch BN层的参数不一样,dropout在训练时随机失效点具有随机性,所以训练和测试要区分开来。 使用时切记要根据实际情况切换: model.train()model.eval() 切记!!!!!!!! 究 阅读全文
posted @ 2019-04-09 22:45 那抹阳光1994 阅读(1032) 评论(0) 推荐(0) 编辑