pytorch/examples

word_language_model with torch.nn.modules.transformer

Open

#1,112 建立於 2023年2月9日

在 GitHub 查看
 (3 留言) (0 反應) (0 負責人)Python (21,634 star) (9,429 fork)batch import
docsgood first issuenlp

描述

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.

貢獻者指南

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