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 (9,429 forks)batch import
docsgood first issuenlp

Repository metrics

Stars
 (21,634 stars)
PR merge metrics
 (No merged PRs in 30d)

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