使用IterableWrapper时未定义WikiText2

Question

我正在学习本教程，https://pytorch.org/tutorials/beginner/transformer_tutorial.html。

在调用此函数时，我将得到以下错误。

/usr/local/lib/python3.7/dist-packages/torchtext/datasets/wikitext2.py in WikiText2(根，拆分) 75 ) 76 --> 77 url_dp = IterableWrapper(URL) 78 #磁盘上的缓存数据79 cache_compressed_dp = url_dp.on_disk_cache(

NameError:未定义名称“IterableWrapper”

以下是代码：

from torchtext.datasets import WikiText2
from torchtext.data.utils import get_tokenizer
from torchtext.vocab import build_vocab_from_iterator
from torchdata.datapipes.iter import IterableWrapper

train_iter = WikiText2(split='train')

如果你有什么想法告诉我..。谢谢

安托万

Answer 1

我试着运行你提供的代码片段。我看不出来

NameError:名称“IterableWrapper”未定义为

但我有个不同的错误说，

无模块名为“torchdata”

我没有安装手电筒。因此，在您的情况下，我将确保火炬数据安装是否正确。你可以看看这个官方的Git，看看它是否能实现https://github.com/pytorch/data

诚挚的问候