python - 如何更改gpt-2代码以与Tensorflow 2.0一起使用？

我正在尝试使用gpt-2进行文本生成。即使在运行Tensorflow 2.0 code upgrade script之后，我仍然遇到兼容性错误。
我遵循的步骤:

克隆repo

从这里开始，按照DEVELOPERS.md中的指示进行操作

在/ src中的文件上运行upgrade script

在终端运行:sudo docker build --tag gpt-2 -f Dockerfile.gpu .

构建完成后，运行:sudo docker run --runtime=nvidia -it gpt-2 bash

输入python3 src/generate_unconditional_samples.py | tee /tmp/samples

获取此追溯:

Traceback (most recent call last):
File "src/generate_unconditional_samples.py", line 9, in <module>
import model, sample, encoder
File "/gpt-2/src/model.py", line 4, in <module>
from tensorboard.plugins.hparams.api import HParam
ImportError: No module named 'tensorboard.plugins.hparams'
root@f8bdde043f91:/gpt-2# python3 src/generate_unconditional_samples.py | tee
/tmp/samples
Traceback (most recent call last):
File "src/generate_unconditional_samples.py", line 9, in <module>
import model, sample, encoder
File "/gpt-2/src/model.py", line 4, in <module>
from tensorboard.plugins.hparams.api import HParam
ImportError: No module named 'tensorboard.plugins.hparams'```

HParams似乎已被弃用，Tensorflow 2.0中的新版本称为HParam。但是，参数不同。在model.py中，参数的实例化如下:

def default_hparams():
return HParams(
    n_vocab=0,
    n_ctx=1024,
    n_embd=768,
    n_head=12,
    n_layer=12,
)

Tensorflow 2.0中似乎没有任何1:1转换。有谁知道如何使gpt-2与Tensorflow 2.0一起使用？
我的GPU是NVIDIA 20xx。
谢谢。

最佳答案

如果要查看我的1.x fork，请在此处进行编译:
https://github.com/timschott/gpt-2

关于python - 如何更改gpt-2代码以与Tensorflow 2.0一起使用？，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/63350105/