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



假设我有一个类似的模块

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

然后我做:

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

更换者是否已登记?接下来会有什么令人讨厌的副作用吗?

是。这是重新分配属性的标准方法。是否有严重的副作用取决于你的替代品是否符合规定。

最新更新