1. Nauka
    2. /
  2. Kursy
    3. /
  3. Głębokie uczenie z PyTorch – poziom średnio zaawansowany
Connected

ćwiczenie

Sieć GRU

Obok LSTM popularnym wariantem rekurencyjnej sieci neuronowej jest jednostka bramkowana GRU (Gated Recurrent Unit). Jej zaletą jest prostota: komórki GRU wymagają mniej obliczeń niż komórki LSTM, często dorównując im jednak pod względem wydajności.

Dostarczony kod zawiera wcześniej zdefiniowany model RNN. Twoim zadaniem jest jego dostosowanie tak, aby uzyskać sieć GRU zamiast RNN. Biblioteka torch oraz torch.nn as nn zostały już zaimportowane.

Instrukcje

100 XP
  • Zaktualizuj definicję modelu RNN, aby otrzymać sieć GRU; przypisz warstwę GRU do self.gru.