1. Učit se
  2. /
  3. Kurzy
  4. /
  5. Apprentissage profond intermédiaire avec PyTorch

Connected

cvičení

Réseau GRU

En plus des LSTM, une autre variante populaire des réseaux de neurones récurrents est l'unité récurrente à portes, ou GRU. Son avantage réside dans sa simplicité : les cellules GRU exigent moins de calcul que les cellules LSTM, tout en offrant souvent des performances comparables.

Le code fourni est la définition du modèle RNN que vous avez programmée plus tôt. Votre tâche consiste à l'adapter pour obtenir un réseau GRU. torch et torch.nn as nn ont déjà été importés pour vous.

Pokyny

100 XP
  • Mettez à jour la définition du modèle RNN afin d'obtenir un réseau GRU; affectez la couche GRU à self.gru.