Encoder.py и train.py в GPT-2 не работают

Я пытаюсь обучить 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

Я пробовал все решения, которые нашел в Интернете и которые мог придумать, но я застрял. Пожалуйста помоги

1 ответ

Проверьте документацию:https://github.com/openai/gpt-2-output-dataset/tree/master/detector

Ожидается, что вы запустите его как python -m detector.train.

Другие вопросы по тегам