36-cannot

¶cannot assign module before Module.__init__() call

在pytorch编写代码时遇到了以下问题


Encoder代码如下

class Encoder(nn.Module):
    def __init__(self, input_channel, embed_dim, batchNorm_momentum):
        super(Encoder, self).__init__()
        self.input_channel = input_channel
        self.embed_dim = embed_dim

¶解决

在Generator中添加父类初始化解决问题

class Generator(nn.Module):
    def __init__(self, input_channel, embed_dim, batchNorm_momentum=0.2):
        super(Generator, self).__init__() ########
        self.input_channel = input_channel
        self.embed_dim = embed_dim