JayYip/m3tl

请问您的项目支持tf1.15吗?

kFoodie opened this issue · 10 comments

您好,非常感谢您的分享。
我在使用您的项目的时候,发现tensor2tensor好像只支持tf2.2以上,然后看您的项目里面又是支持tf1.13;我装了tf1.15之后,运行您demo里面的代码,报了错

File "/root/data/glusterfs_sharing_04_v3/11117720/bert-multitask-learning-master/bert_multitask_learning/run_bert_multitask.py", line 120, in train_bert_multitask
input_fn=train_input_fn, max_steps=params.train_steps, hooks=[train_hook])
File "/root/miniconda3/envs/tf1.15/lib/python3.6/site-packages/tensorflow_estimator/python/estimator/training.py", line 163, in new
'Must specify max_steps > 0, given: {}'.format(max_steps))
ValueError: Must specify max_steps > 0, given: 0

使用的是这个代码 https://github.com/JayYip/bert-multitask-learning/blob/master/notebooks/Run%20Pre-defined%20problems.ipynb
希望您能帮忙解答一下,万分感谢

您的项目中使用了tensorflow.contrib,然后相关依赖使用了tf2.2.0,最后会报No module named 'tensorflow.contrib'。。

0.3.x支持1.15,但是现在安装比较麻烦... 需要自行确定tf1支持的t2t和dopamine-rl的版本

您的项目中使用了tensorflow.contrib,然后相关依赖使用了tf2.2.0,最后会报No module named 'tensorflow.contrib'。。

是,部分代码尚未完全迁移,用到contrib模块主要是crf和export_model.py

感谢您的回复

现在您的代码在tf2.2.0上是可以正常使用了吗?

是的,不过notebook还没有更新

难怪我这里一直报错……期待您的更新