model.eval()

作者: 啊啊啊啊啊1231 | 来源:发表于2019-12-26 10:08 被阅读0次

把模型model.eval()作用是为了固定BN和dropout层,使得偏置参数不随着发生变化。因为当batchsize小时,如果没有固定,会对图像的失真有很大的影响。

相关文章

网友评论

    本文标题:model.eval()

    本文链接:https://www.haomeiwen.com/subject/eurkoctx.html