训练需要
model.train()
:启用 BatchNormalization 和 Dropout
测试必要
(pytorch是默认开启BatchNormalization 和 Dropout的,测试或者pre-train的时候必须使用)
model.eval()
:不启用 BatchNormalization 和 Dropout
训练需要
model.train()
:启用 BatchNormalization 和 Dropout
测试必要
(pytorch是默认开启BatchNormalization 和 Dropout的,测试或者pre-train的时候必须使用)
model.eval()
:不启用 BatchNormalization 和 Dropout