[Pytorch] nn.Module을 상속받을 때 super().__init__()을 하는 이유

qw4735·2022년 12월 19일
0

PyTorch

목록 보기
4/8

torch.nn.Module

  • Base class for all neural network modules.
  • Your models should also subclass this class.

super().__init__()이란?

  • super() : 상속받은 부모클래스를 의미
  • 부모 클래스를 불러와서 __init__ ()을 수행한다는 것은 부모클래스의 생성자를 불러주는 것을 의미
  • super().__init__()은 nn.Module의 __init__()에 선언되어 있는 변수들을 상속받는 역할을 함

\therefore 결론적으로 torch.nn의 모듈을 사용하기 위해서는 반드시 super.__init__()을 해줘야함
reference : https://cnu-jinseop.tistory.com/185?category=998353

0개의 댓글