torch.nn.Module中的training属性详情，与Module.train()和Module.eval（）的关系

最新推荐文章于 2025-02-19 09:42:34 发布

chaiiiiiiiiiiiiiiiii

最新推荐文章于 2025-02-19 09:42:34 发布

阅读量1.1w

点赞数 14

分类专栏： python代码有关文章标签： python 深度学习

本文链接：https://blog.youkuaiyun.com/qq_37959202/article/details/105148135

版权

Module类的构造函数：

    def __init__(self):
        """
        Initializes internal Module state, shared by both nn.Module and ScriptModule.
        """
        torch._C._log_api_usage_once("python.nn_module")

        self.training = True
        self._parameters = OrderedDict