Pytorch WikiText2 不适用于我的本地笔记本,但适用于 Google Colab

我在这里学习本教程: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'
z145071 回答:Pytorch WikiText2 不适用于我的本地笔记本,但适用于 Google Colab

暂时没有好的解决方案,如果你有好的解决方案,请发邮件至:iooj@foxmail.com
本文链接:https://www.f2er.com/36594.html

大家都在问