[Pytorch] torch.nn.Sequential

class torch.nn.Sequential(*args)[source]

편하게 순차적으로 실행하도록 담는 container라고 생각하면 될 것 같다.

내가 했던것을 예를들면 

x = self.conv1(x)

x = self.dropout(x)

x = self.pool(F.relu(x))

x = self.conv2(x)

... 

뭐 이런식으로 썼다. 

하지만 위의 함수를 쓰면 

model = nn.Sequential(

nn.Conv2d(....),

nn.ReLU(),

nn.Conv2d(20,64,5),

nn.RELU()

)

이므로 forward() 에서 

x = self.model(x) 만 써주면 된다.

깔끔하고 편리한 함수인것 같다.

이걸 왜 지금 알았을까 

원문보러가기