pytorch/examples

word_language_model with torch.nn.modules.transformer

Open

Aperta il 9 feb 2023

Vedi su GitHub
 (3 commenti) (0 reazioni) (0 assegnatari)Python (21.634 star) (9429 fork)batch import
docsgood first issuenlp

Descrizione

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.

Guida contributor