Začněte nyníZačněte zdarma

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

Zobrazit kurz

Pokyny k cvičení

  • Definuj transformer s 8 attention heads, 6 vrstvami enkodéru a dekodéru a vstupními embeddingy sekvencí o délce 1536.
  • 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(____)
Upravit a spustit kód