【发布时间】:2020-01-03 08:47:58
【问题描述】:
我正在尝试重现本文中的示例:https://medium.com/@ngwaifoong92/beginners-guide-to-retrain-gpt-2-117m-to-generate-custom-text-content-8bb5363d8b7f
示例代码来自以下仓库:https://github.com/nshepperd/gpt-2
安装需求并下载模型后,接下来是训练模型,需要执行此代码:
python encode.py lyric.txt lyric.npz
这里的问题是这需要导入以下模块:
import argparse
import numpy as np
import encoder
from load_dataset import load_dataset
encoder 和 load_dataset 位于子目录中:
|--encode.py
--src
|--encoder.py
|--load_dataset.py
这会产生以下错误:
ModuleNotFoundError: No module named 'encoder'
我尝试创建__init__.py 文件并将它们导入为
src.encoder 和 src.load_dataset 但它们也不起作用。
在中篇文章中,作者建议将文件 encoder.py 移动到 src 并从那里执行代码,问题是这样做也会破坏模型的相对路径,尽管我处理了其他文件的路径问题。
【问题讨论】:
-
如果成功请接受,谢谢
标签: python path nlp init gpt-2