1. Learn
    2. /
  2. Cursuri
    3. /
  3. Deep Learning intermediar cu PyTorch
Connected

exercițiu

Bucla de antrenament pentru RNN

E momentul să antrenezi modelul de prognoză a consumului de electricitate!

Vei folosi rețeaua LSTM definită anterior, care a fost instanțiată și atribuită variabilei net, la fel ca și dataloader_train construit înainte. Va trebui să utilizezi și torch.nn, deja importat sub numele nn.

În acest exercițiu, vei antrena modelul timp de doar trei epoci, pentru a verifica dacă procesul de antrenament decurge corect. Să începem!

Instrucțiuni

100 XP
  • Configurează funcția de pierdere Mean Squared Error și atribuie-o variabilei criterion.
  • Remodelează seqs la forma (batch size, sequence length, num features), adică (32, 96, 1), și reatribuie rezultatul variabilei seqs.
  • Pasează seqs modelului pentru a obține outputs.
  • Pe baza valorilor calculate anterior, calculează pierderea și atribuie-o variabilei loss.