1. Learn
  2. /
  3. Courses
  4. /
  5. Intermediate Deep Learning with PyTorch

Connected

Exercise

Síť GRU

Vedle LSTM sítí patří mezi oblíbené varianty rekurentních neuronových sítí také Gated Recurrent Unit, neboli GRU. Jeho hlavní předností je jednoduchost: GRU buňky vyžadují méně výpočetních prostředků než LSTM buňky, přičemž jim výkonností velmi často vyrovnají.

Níže máš k dispozici definici modelu RNN, kterou jsi napsal/a v předchozím cvičení. Tvojím úkolem je upravit ji tak, aby vznikla síť GRU. torch a torch.nn as nn jsou již naimportovány.

Instructions

100 XP
  • Uprav definici modelu RNN tak, aby vznikla síť GRU; GRU vrstvu přiřaď do self.gru.