1. Learn
    2. /
  2. Cursuri
    3. /
  3. Deep Learning intermediar cu PyTorch
Connected

exercițiu

Rețea GRU

Pe lângă LSTM-uri, o altă variantă populară de rețea neuronală recurentă este Gated Recurrent Unit, sau GRU. Atractivitatea sa constă în simplitate: celulele GRU necesită mai puțin calcul decât celulele LSTM, obținând adesea performanțe similare.

Codul furnizat reprezintă definiția modelului RNN pe care l-ai implementat anterior. Sarcina ta este să îl adaptezi astfel încât să producă o rețea GRU în loc. torch și torch.nn as nn au fost deja importate pentru tine.

Instrucțiuni

100 XP
  • Actualizează definiția modelului RNN pentru a obține o rețea GRU; atribuie stratul GRU variabilei self.gru.