model.train():
启用 batch normalization 和 dropout ,当你的网络中有bn层,在训练时就应启用model.train()
model.eval():
不启用batch normalization 和 dropout ,防止测试时参数改变
model.train():
启用 batch normalization 和 dropout ,当你的网络中有bn层,在训练时就应启用model.train()
model.eval():
不启用batch normalization 和 dropout ,防止测试时参数改变