GPT-2 的 encoder.py 和 train.py 不起作用

问题描述

我正在尝试训练 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

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

解决方法

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

您应该以 python -m detector.train 的身份运行它。