gpt4 book ai didi

pytorch - torch.nn 模块中函数 eval() 的含义是什么

转载 作者:行者123 更新时间:2023-12-02 09:43:27 26 4
gpt4 key购买 nike

官方评论显示“这仅对 Dropout 或 BatchNorm 等模块有任何影响。”但我不明白它的实现。

最佳答案

Dropout 和 BatchNorm(可能还​​有一些自定义模块)在训练和评估期间表现不同。您必须通过在模型上调用 .eval() 让模型知道何时切换到 eval 模式。

这会将模型中每个模块的 self.training 设置为 False。如果您正在实现自己的模块,并且在训练和评估期间必须表现不同,则可以在执行此操作时检查 self.training 的值。

关于pytorch - torch.nn 模块中函数 eval() 的含义是什么,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/48146926/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com