为什么我的 PyTorch ModuleList object 在分配为 class 变量后变成了一个“函数”object？

Question

我真的需要你的帮助来解决一些我无法解释的烦人问题。 我是 PyTorch 的新手，想实现一个需要使用 ModuleList 的模块。 但是，在实例化一个空的 ModuleList object 之后，我无法使用它的追加方法。 为什么？

self.modules = torch.nn.ModuleList([])
print(type(self.modules), type(torch.nn.ModuleList())) # -> <class 'method'> <class 'torch.nn.modules.container.ModuleList'>
for m in range(len(self.module_sizes)):

        prev_channels = 0 if m == 0 else self.module_out_channels[-m]
        vgg19_channels = self.vgg19_layer_channels[-(m+1)]
        module_out_channels = self.module_out_channels[-(m+1)]
        module_size = self.module_sizes[-(m+1)]

        self.modules.append(
            GeneratorModule(prev_channels=prev_channels, vgg19_channels=vgg19_channels, out_channels=module_out_channels, module_size=module_size, initial=(m==0))
            )

错误消息说：

  File "main.py", line 90, in __init__
    self.modules.append(
AttributeError: 'function' object has no attribute 'append'

到目前为止，一切都很好。 我假设一旦我将 ModuleList 作为 class 变量分配给我的自定义 nn.Module 子类，它就会变成一个“函数”object。所以我更改了代码，在附加后将 ModuleList 分配给 self.modules环形。 但随后它在后面的一段代码中抛出了另一个错误，它说：

for m, module in enumerate(self.modules):
     pass

TypeError: 'method' object is not iterable

我知道这应该可行，因为我发现其他人在 inte.net 上也是这样做的。

如果有更多经验丰富的 PyTorch 程序员可以帮助我，我将非常高兴。 提前致谢！

Answer 1

其实问题是不一样的。 如您所见， torch.nn.ModuleList具有append和其他类似列表的方法，因此它应该可以工作。 不，它不会“改变”为“功能”object，这对每个人来说都是非常不直观的。

相反，PyTorch 的torch.nn.Module有一些你继承的方法，其中包括modules （它允许你迭代包含在torch.nn.Module实例中的modules ）。

只需将您的self.modules重命名为self.module_list就可以了。