我们在训练时如果使用了BN层和Dropout层,我们需要对model进行标识:
model.train():在训练时使用BN层和Dropout层,对模型进行更改。
model.eval():在评价时将BN层和Dropout层冻结,这两个操作不会对模型进行更改。