Zacznij terazZacznij za darmo

Transformery w PyTorch

Skoro znasz już różne komponenty architektury transformera, czas zdefiniować własny model! Moduł torch.nn, zaimportowany jako nn, umożliwia zrobienie tego w zaledwie kilku linijkach kodu.

To ćwiczenie jest częścią kursu

Modele Transformer w PyTorch

Zobacz kurs

Instrukcje do ćwiczenia

  • Zdefiniuj transformer z 8 głowicami uwagi, 6 warstwami enkodera i dekodera oraz osadzeniami sekwencji wejściowej o długości 1536.
  • Wyświetl obiekt modelu, aby zobaczyć jego architekturę.

Interaktywne ćwiczenie praktyczne

Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.

# Define the transformer model
model = ____

# Print the model object
print(____)
Edytuj i uruchom kod