我在这里学习本教程:https://pytorch.org/tutorials/beginner/transformer_tutorial.html
在 Google colab 代码上一切正常,但问题是我下载了 Jupyter Notebook 以在我的本地机器上运行,而完全相同的代码不起作用。
在我的本地机器上,我用 Conda 创建了一个新的 env,并安装了最新版本的 Pytorch、torchtext 等。当我到达这部分时,它失败并出现以下错误:
代码块:
import torch
from torchtext.datasets import WikiText2
from torchtext.data.utils import get_tokenizer
from torchtext.vocab import build_vocab_from_iterator
train_iter = WikiText2(split='train')
tokenizer = get_tokenizer('basic_english')
vocab = build_vocab_from_iterator(map(tokenizer,train_iter),specials=["<unk>"])
vocab.set_default_index(vocab["<unk>"])
错误:
---------------------------------------------------------------------------
TypeError Traceback (most recent call last)
<ipython-input-8-a461d2b962de> in <module>
4 from torchtext.vocab import build_vocab_from_iterator
5
----> 6 train_iter = WikiText2(split='train')
7 tokenizer = get_tokenizer('basic_english')
8 vocab = build_vocab_from_iterator(map(tokenizer,specials=["<unk>"])
TypeError: __init__() missing 2 required positional arguments: 'path' and 'text_field'