1. 学ぶ
    2. /
  2. コース
    3. /
  3. PyTorch로 배우는 딥러닝 입문
Connected

演習

Dropout 실험해 보기

Dropout은 학습 중에 일부 출력 값을 무작위로 0으로 만들어 과적합을 줄이는 데 도움을 줍니다. 이 연습에서는 dropout이 포함된 간단한 신경망을 만들고, 학습 모드와 평가 모드에서 어떻게 동작하는지 살펴보세요.

torch.nn 패키지는 nn으로 미리 로드되어 있으며, features도 이미 정의되어 있어요.

指示1 / 2

undefined XP
    1
    2
  • 요소를 50% 확률로 0으로 만드는 dropout 레이어를 모델에 추가하세요.
  • 모델을 학습 모드로 설정하고 features를 사용해 예측을 생성한 뒤, 출력을 output_train으로 저장하세요.