New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

no module named tensorflow.contrib #122

Open

sprajagopal opened this issue Jun 1, 2021 · 1 comment

sprajagopal commented Jun 1, 2021 •

edited

Loading

python3.8
transformers==4.4.1
tensorflow==2.5.0

On running:

python run_pretraining.py --data-dir data --model-name electra_small

it exits with the following error:

   from tensorflow.contrib import layers as contrib_layers
ModuleNotFoundError: No module named 'tensorflow.contrib'

Is this issue related to tf1 vs tf2 or is this something else?

The text was updated successfully, but these errors were encountered:

nbroad1881 commented Jul 31, 2021

https://github.com/google-research/electra#requirements

TensorFlow 1.15 (although we hope to support TensorFlow 2.0 at a future date)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment