gpt-2 修改为适用于tensorflow 2.0

2hh7jdfx  于 6个月前  发布在  其他
关注(0)|答案(4)|浏览(95)

由于我正在运行Arch,处理旧版本的软件绝对是一件痛苦的事情。因此,我想将其更新到tensorflow 2.0。最大的问题是在model.py脚本中,因为它依赖于tensorflow.contrib,而tensorflow.contrib已经不存在(即使在tensorflow.compat.v1中)。我如何轻松修改此脚本以使其与最新最好的tensorflow一起工作?

x9ybnkn6

x9ybnkn61#

在Debian(Raspbian)Buster上运行TF 2.1时也有同样的问题。我希望这个问题能很快得到解决。

nszi6y05

nszi6y052#

大家好,请看一下TFGPT 2 Model:https://huggingface.co/transformers/v2.0.0/model_doc/gpt2.html部分中实现的lib,并考虑升级TF:https://www.tensorflow.org/guide/upgrade
On Fri,Feb 21,2020,02:57 Dave Xanatos***@***.***> wrote:Having same issue with tensorflow. contrib. Running TF 2.1 on Debian(Raspbian)巴斯特.我希望这可以尽快解决. -你收到这个,因为你订阅了这个线程.直接回复这封电子邮件,在GitHub上查看<#231?email_source=notifications&email_token= AEYAMLZZALQG 5 HF 3VYLOLMLRD 3 OB 3A 5CNFSM 4KYVERL2YY3PNVWWK3TUL52HS4DFVREXG43VMVBW63LNMVXHJKTDN5WW2ZLOORPWSZGOEMP3AUA#issuecomment-589279312>,或取消订阅< https://github.com/notifications/unsubscribe-auth/AEYAMLZ4OPDSPGIOKMDPNJLRD3OB3ANCNFSM4KYVERLQ >。

gudnpqoy

gudnpqoy3#

有没有人找到在TF 2.2中运行这个的方法?

ncgqoxb0

ncgqoxb04#

不需要担心整个tensorflow.contrib模块。在这个项目中,只使用了来自tensorflow.contrib.trainingHParams类(在model.py中导入)。
我发现(tensorflow/community#148)原来的class可以替换为一个漂亮的fork。将hparam.py保存在src文件夹中,并将导入替换为

from hparam import HParams

解决了原来的问题。

UPD:刚刚出现了一个pull request #262,将关闭此问题。

相关问题