1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Середній рівень Deep Learning з PyTorch
Connected

cvičení

Мережа GRU

Поряд із LSTM, ще одним популярним різновидом рекурентних нейронних мереж є Gated Recurrent Unit, або GRU. Її привабливість — у простоті: комірки GRU потребують менше обчислень, ніж комірки LSTM, водночас часто не поступаючись їм за якістю.

Наданий вам код — це визначення моделі RNN, яке ви писали раніше. Ваше завдання — адаптувати його так, щоб замість цього створити мережу GRU. torch і torch.nn as nn вже імпортовано.

Pokyny

100 XP
  • Оновіть визначення моделі RNN, щоб отримати мережу GRU; присвойте шар GRU до self.gru.