Transformery v PyTorchi
Teď, když znáš jednotlivé komponenty architektury transformeru, je čas ji sestavit! Modul torch.nn, importovaný jako nn, nabízí elegantní způsob, jak to zvládnout na pár řádků kódu.
Toto cvičení je součástí kurzu
Transformer Models with PyTorch
Pokyny k cvičení
- Definuj transformer s
8attention heads,6vrstvami enkodéru a dekodéru a vstupními embeddingy sekvencí o délce1536. - Vypiš objekt modelu, aby ses podíval/a na jeho architekturu.
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
# Define the transformer model
model = ____
# Print the model object
print(____)