[파이토치] nn.Sequential 관련

 

 

 

 

 

파이토치에 있는 Sequential 클래스는 각각의 모듈이 순차적으로 지나갈 수 있도록 구조를 정의할 수 있다. 예제 샘플을 확인하면 쉽게 이해할 수 있다.

 

 

Sample code

# Example of using Sequential
model = nn.Sequential(
          nn.Conv2d(1,20,5),
          nn.ReLU(),
          nn.Conv2d(20,64,5),
          nn.ReLU()
        )

# Example of using Sequential with OrderedDict
model = nn.Sequential(OrderedDict([
          ('conv1', nn.Conv2d(1,20,5)),
          ('relu1', nn.ReLU()),
          ('conv2', nn.Conv2d(20,64,5)),
          ('relu2', nn.ReLU())
        ]))

 

 

https://pytorch.org/docs/stable/nn.html

torch.nn — PyTorch master documentation