python - GPT-2 的 encoder.py 和 train.py 不工作

Question

我正在尝试训练 GPT-2 使用我在文本文件 napoleon.txt 中提供的内容。当我运行编码器时，它似乎可以在命令提示符下工作。

python encoder.py napoleon.txt napoleon.npz

然而，它实际上并没有创建 napoleon.npz。但这只是问题的一部分。更大的问题是 train.py，我训练 GPT-2 真正需要的，每次都会吐出一个错误。

Traceback (most recent call last):
  File "train.py", line 19, in <module>
    from .dataset import Corpus, EncodedDataset
ImportError: attempted relative import with no known parent package

我已经尝试了我在互联网上找到的所有我能想到的解决方案，但我被困住了。请帮忙

Answer 1

检查文档： https ://github.com/openai/gpt-2-output-dataset/tree/master/detector

您应该将其作为python -m detector.train.