목록nn.Sequential (1)

무슨 생각을 해 그냥 하는거지

[Pytorch] nn.Sequential은 multiple input을 다룰 수 없다

기초적인 얘기지만 혹시나 나중에 까먹을까봐 + 나처럼 답답해 하고 있는 분이 계실까봐 기록한다. 분명 forward 때 인풋을 두 개 줬는데 자꾸 에러가 발생했다. *** TypeError: forward() takes 2 positional arguments but 3 were given 다른 부분은 이상이 없어보였는데, 찾아보니 nn.Sequential이 문제였다. nn.Sequential로 선언한 레이어는 forward 때 인풋을 하나만 넣어줄 수 있다. (나는 Encoder Layer를 nn.Sequential 안에 여러 개 넣어놨는데 Encoder layer의 forward가 2개의 인자를 받아야 했다.) 예를 들어.. 어떤 모델 클래스 __init__ 함수에서 이렇게 선언했다면 self.enc..
COMPUTER/Deep Learning 2022. 2. 16. 11:22
이전 Prev 1 Next 다음