pytorch/examples

word_language_model with torch.nn.modules.transformer

Open

#1.112 geöffnet am 9. Feb. 2023

Auf GitHub ansehen
 (3 Kommentare) (0 Reaktionen) (0 zugewiesene Personen)Python (21.634 Stars) (9.429 Forks)batch import
docsgood first issuenlp

Beschreibung

The torch.nn.modules.transformer documentation says the word_language_model example in this repo is an example of its use. But it seems to instead DIY a transformer and uses that instead. Is this intentional? I would offer my help to write it for torch.nn.modules.transformer but I'm here to learn how to use it.

Contributor Guide