Mencoba dropout
Dropout membantu mencegah overfitting dengan secara acak mengatur beberapa nilai keluaran menjadi nol selama pelatihan. Pada latihan ini, Anda akan membangun jaringan saraf sederhana dengan dropout dan mengamati perilakunya dalam mode pelatihan dan evaluasi.
Paket torch.nn sudah dimuat sebagai nn, dan features sudah disediakan untuk Anda.
Latihan ini adalah bagian dari kursus
Pengantar Deep Learning dengan PyTorch
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# Model with Dropout
model = nn.Sequential(
nn.Linear(8, 6),
nn.Linear(6, 4),
____)
# Forward pass in training mode (Dropout active)
model.____
output_train = ____