새로운 합성곱 계층 추가하기
프로젝트 리드가 새 CNN 모델을 제공했어요. 모델의 아키텍처를 살펴보고 여기에 새로운 합성곱 계층을 추가해 보세요.
모델은 CNNModel로 제공돼요. 패키지 torch와 torch.nn(별칭 nn)은 이미 임포트되어 있어요.
이 연습은 강의의 일부입니다
PyTorch로 배우는 이미지 딥러닝
연습 안내
CNNModel클래스로부터 모델을 인스턴스화하고 합성곱 계층들에 접근하세요.- 기존 계층의
out_channels를 새로운 계층의in_channels로 하고,out_channels는 32,stride와padding은 각각 1,kernel_size는 3으로 설정해 새 합성곱 계층을 만들고 이를conv2에 할당하세요. - 새 계층을 모델에
"conv2"라는 이름으로 추가하세요.
실습형 인터랙티브 연습
이 예제를 이 샘플 코드를 완성하여 풀어보세요.
# Create a model
model = ____
print("Original model: ", model)
# Create a new convolutional layer
conv2 = ____
# Append the new layer to the model
model.____(____)
print("Extended model: ", model)