Dropout 실험해 보기
Dropout은 학습 중에 일부 출력 값을 무작위로 0으로 만들어 과적합을 줄이는 데 도움을 줍니다. 이 연습에서는 dropout이 포함된 간단한 신경망을 만들고, 학습 모드와 평가 모드에서 어떻게 동작하는지 살펴보세요.
torch.nn 패키지는 nn으로 미리 로드되어 있으며, features도 이미 정의되어 있어요.
이 연습은 강의의 일부입니다
PyTorch로 배우는 딥러닝 입문
실습형 인터랙티브 연습
이 예제를 이 샘플 코드를 완성하여 풀어보세요.
# Model with Dropout
model = nn.Sequential(
nn.Linear(8, 6),
nn.Linear(6, 4),
____)
# Forward pass in training mode (Dropout active)
model.____
output_train = ____