search

allenai / bilm-tf

Tensorflow implementation of contextualized word representations from bi-directional language models
Apache License 2.0
1.62k stars 452 forks source link

Upgrade to support TensorFlow 2.x #222

Open InternetExplorer9 opened 4 years ago

pzhang84 commented 2 years ago

Great work! Can you also share the version of CUDA and TensorFlow you are using?

I tried CUDA11.2 and python=3.6.13 with 1) tensorflow-2.0.0, 2) tensorflow-2.1.0, 3) tensorflow-2.2.0. However, all those settings failed testing pass.

eduamf commented 2 years ago

If you see the changes, most of them are setting the version compatibility with v1, using tf.compat.v1. instead of tf.. Then, it is expected to work from Tensorflow 1.15 till 2.6.0.