Pytorch本人疑问(2)model.train()和model.eval()的区别

我们在训练时如果使用了BN层和Dropout层,我们需要对model进行标识:

model.train():在训练时使用BN层和Dropout层,对模型进行更改。

model.eval():在评价时将BN层和Dropout层冻结,这两个操作不会对模型进行更改。

posted @ 2020-02-27 21:24  谦谦谦谦谦谦谦谦  阅读(...)  评论(...编辑  收藏