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
Instrukcje do ćwiczenia
- Zdefiniuj transformer z
8głowicami uwagi,6warstwami enkodera i dekodera oraz osadzeniami sekwencji wejściowej o długości1536. - 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(____)