[error] AttributeError: cannot assign module before Module.__init__() call 해결

pytorch에서 nn.Module을 상속한 클래스를 만들 때, init을 해주지 않아서 발생하는 오류

 

class aa(nn.Module):
    def __init__(self):
    	super(aa, self).__init__() ## 이거 추가
        self.a = 1
        self.b = 2

 

super()추가하여 수정