Pytorch中的 model.train() 和 model.eval() 模式

2024年4月28日 485次阅读来源: pytorch

model.train() ：启用 BatchNormalization 和 Dropout

model.eval() ：不启用 BatchNormalization 和 Dropout

参考：

    原文作者：pytorch
    原文地址: https://www.cnblogs.com/sbj123456789/p/10825685.html
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。