1. Learn
    2. /
  2. Курси
    3. /
  3. Apprentissage profond intermédiaire avec PyTorch
Connected

вправа

Créer un RNN de prévision

Il est temps de bâtir votre premier réseau récurrent ! Ce sera un modèle séquence-vers-vecteur composé d'une couche RNN avec deux couches et un hidden_size de 32. Après la couche RNN, une simple couche linéaire fera la correspondance des sorties vers une valeur unique à prédire.

Les importations suivantes ont déjà été faites pour vous :

import torch
import torch.nn as nn

Інструкції 1/4

undefined XP
    1
    2
    3
    4
  • Définissez la couche RNN en lui passant les bonnes valeurs pour input_size, hidden_size, num_layers et batch_first, et assignez-la à self.rnn.