pytorch/examples

word_language_model with torch.nn.modules.transformer

Open

#1.112 aberto em 9 de fev. de 2023

Ver no 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.

Guia do colaborador