python - pytorch 是否注册分配给对象属性的模块？

Question

假设我有一个像这样的模块:

class MyModule(nn.Module):
    def __init__(self):
        super().__init__()
        self.cnn = CNN(params)

然后我做:

module = MyModule()
module.cnn = CNN(some_other_params)

替代品是否已注册？会不会有任何严重的副作用？

Answer 1

是的。这是重新分配属性的标准方法。是否存在严重的副作用取决于您的替代品是否兼容指定。