pytorch/examples

word_language_model with torch.nn.modules.transformer

Open

#1,112 opened on Feb 9, 2023

View on GitHub
 (3 comments) (0 reactions) (0 assignees)Python (21,634 stars) (9,429 forks)batch import
docsgood first issuenlp

Description

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

word_language_model with torch.nn.modules.transformer · pytorch/examples#1112 | Good First Issue