1. Learn
    2. /
  2. Kurser
    3. /
  3. Fördjupad djupinlärning med PyTorch
Connected

övning

GRU-nätverk

Vid sidan av LSTM-nätverk finns en annan populär variant av rekurrenta neurala nätverk: Gated Recurrent Unit, eller GRU. Dess styrka ligger i enkelheten – GRU-celler kräver mindre beräkning än LSTM-celler och matchar ändå ofta deras prestanda.

Koden du har tillgång till är den RNN-modelldefinition som du kodade tidigare. Din uppgift är att anpassa den så att den producerar ett GRU-nätverk istället. torch och torch.nn as nn har redan importerats åt dig.

Instruktioner

100 XP
  • Uppdatera RNN-modelldefinitionen för att få ett GRU-nätverk; tilldela GRU-lagret till self.gru.