1. Learn
    2. /
  2. Cursuri
    3. /
  3. Introducere în Deep Learning cu PyTorch
Connected

exercițiu

Experimentând cu dropout

Dropout ajută la prevenirea supraantrenării prin setarea aleatorie a unor valori de ieșire la zero în timpul antrenării. În acest exercițiu, vei construi o rețea neuronală simplă cu dropout și vei observa cum se comportă în modul de antrenare și în cel de evaluare.

Pachetul torch.nn este preîncărcat ca nn, iar features este deja definit pentru tine.

Instrucțiuni 1/2

undefined XP
    1
    2
  • Adaugă un strat de dropout modelului cu o probabilitate de 50% de a seta elemente la zero.
  • Setează modelul în modul de antrenare și generează predicții folosind features, stocând rezultatul ca output_train.