pytorch/examples

word_language_model with torch.nn.modules.transformer

Open

#1 112 ouverte le 9 févr. 2023

Voir sur GitHub
 (3 commentaires) (0 réactions) (0 assignés)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.

Guide contributeur