Python pytorch nn.module如何保存子模块

我对pytorch nn.module的工作原理有一些疑问

import torch
import torch.nn as nn



class Net(nn.Module):
    def __init__(self):
        super(Net, self).__init__()
        self.sub_module = nn.Linear(10, 5)
        self.value = 3

net = Net()
print(net.__dict__)

输出

{'_modules': OrderedDict([('sub_module', Linear (10 -> 5))]),  'value': 3, ...}

我知道类的每个属性都应该存储在\uuuuu dict\uuuuu中，为什么值（int值）在其中，而子模块（nn.module）不是，相反，子模块存储在\umodules

我读了nn.Module实现的代码，但我没有弄明白。有人有什么想法吗

---

谢谢

我会尽量保持简单

每次在类

Net

中创建一个新项时，例如：

self.sub\u module=nn.Linear（10，5）

它调用其父类的方法

\uuuuu setattr\uuuu

，在本例中是

nn.module

。然后，在

\uuuu setattr\uuuu

方法中，参数被存储到它们所属的dict中。在这种情况下，由于

nn.Linear

是一个模块，它被存储到

\u模块中

下面是在
模块
类中执行此操作的一段代码