1. Nauka
    2. /
  2. Kursy
    3. /
  3. Uczenie głębokie dla tekstu z PyTorch
Connected

ćwiczenie

Generowanie tekstu za pomocą RNN – trening i generowanie

Zespół PyBooks chce teraz, abyś wytrenował i przetestował model RNN, który przewiduje kolejny znak w sekwencji na podstawie podanego wejścia – do autouzupełniania tytułów książek. Ten projekt pomoże zespołowi rozwijać modele do uzupełniania tekstu.

Instancja model klasy RNNmodel jest już dla ciebie załadowana. Zmienna data została wstępnie przetworzona i zakodowana jako sekwencja.

Zmienne inputs i targets są również załadowane.

Instrukcje

100 XP
  • Utwórz instancję funkcji straty, która posłuży do obliczania błędu modelu.
  • Utwórz instancję optymalizatora z modułu optymalizacji PyTorch.
  • Uruchom proces treningu modelu: ustaw model w tryb treningowy i wyzeruj gradienty przed wykonaniem kroku optymalizacji.
  • Po zakończeniu treningu przełącz model w tryb ewaluacji, aby przetestować go na przykładowym wejściu.