pytorch model.train()

model.train():

在使用pytorch构建神经网络的时候,训练过程中会在程序上方添加一句model.train(),作用是启用batch normalization和drop out。

model.eval():

测试过程中会使用model.eval(),这时神经网络会沿用batch normalization的值,并不使用drop out。

torch.no_grad():

你可能感兴趣的:(pytorch)