神经网络

model.eval()

2019-12-26  本文已影响0人  啊啊啊啊啊1231

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

上一篇 下一篇

猜你喜欢

热点阅读