1. Učit se
  2. /
  3. Kurzy
  4. /
  5. Deep Learning for Text with PyTorch

Connected

cvičení

Generování textu pomocí RNN – trénování a generování

Tým PyBooks teď potřebuje natrénovat a otestovat model RNN, který je navržený tak, aby předpovídal další znak v sekvenci na základě zadaného vstupu – a umožnil tak automatické doplňování názvů knih. Tento projekt pomůže týmu dále rozvíjet modely pro doplňování textu.

Instance model třídy RNNmodel je pro tebe předem načtena. Proměnná data byla předzpracována a zakódována jako sekvence.

Proměnné inputs a targets jsou také předem načteny.

Pokyny

100 XP
  • Vytvoř instanci funkce ztráty, která se použije k výpočtu chyby modelu.
  • Vytvoř instanci optimalizátoru z optimalizačního modulu PyTorche.
  • Spusť trénování modelu: přepni ho do trénovacího režimu a před krokem optimalizace vynuluj gradienty.
  • Po dokončení trénování přepni model do režimu vyhodnocení a otestuj ho na vzorovém vstupu.