1. 학습
  2. /
  3. 강의
  4. /
  5. PyTorch로 배우는 이미지 딥러닝

Connected

연습 문제

새로운 합성곱 계층 추가하기

프로젝트 리드가 새 CNN 모델을 제공했어요. 모델의 아키텍처를 살펴보고 여기에 새로운 합성곱 계층을 추가해 보세요.

모델은 CNNModel로 제공돼요. 패키지 torch와 torch.nn(별칭 nn)은 이미 임포트되어 있어요.

지침

100 XP
  • CNNModel 클래스로부터 모델을 인스턴스화하고 합성곱 계층들에 접근하세요.
  • 기존 계층의 out_channels를 새로운 계층의 in_channels로 하고, out_channels는 32, stride와 padding은 각각 1, kernel_size는 3으로 설정해 새 합성곱 계층을 만들고 이를 conv2에 할당하세요.
  • 새 계층을 모델에 "conv2"라는 이름으로 추가하세요.