fastai와 파이토치가 만나 꽃피운 딥러닝

676

4



밑바닥부터 구현하는 딥러닝

이제 이렇게 만든 조각들을 조합하여

CNN

신경망을 만들어보죠.

19.2.1



CNN

앞서 본 대로

Sequential

을 사용하면 여러 구조를 쉽게 구현할 수 있습니다.

Sequential

도

직접 만들어보겠습니다.

class Sequential(Module):

def __init__(self, *layers):

super().__init__()

self.layers = layers

self.register_modules(*layers)

def forward(self, x):

for l in self.layers: x = l(x)

return x

forward

메서드는 단지 각 계층을 순서대로 호출합니다.

Module

의

register

_

modules

메서

드를 사용한다는 점에 유의하세요. 그래야만

parameters

메서드 호출 시 각 모듈의 파라미터

가 모두 나타납니다.

NOTE

_



이 장에서는 모듈 정의에 파이토치 기능을 전혀 사용하지 않습니다. 모든 부분을 우리 스스로 정의하죠.

register

_

modeuls

의 역할과 필요성을 잘 모르겠다면,

Module

클래스의 구현을 보고 스스로 파악해보세요.

적응 풀링을 간소화한

AdaptivePool

모듈을 정의합니다. 이 모듈을 호출하면 입력 ...