Tensorflow error with BERT model selector


I am have created a fresh Python 3 conda environment to use with Knime on Windows 10. When I try to execute the BERT Model Selector, it fails with the error “cannot import name ‘parameter_server_strategy_v2’ from ‘tensorflow.python.distribute’” see screenshot below.

I’ve checked the packages required in the BERT Knime Hub article and all are installed at equal or higher version (for example tensorflow-hub is 0.12). Any idea why I would be getting this error?


Managed to resolve the issue - tensorflow estimator was version 2.4 - needed to downgrade to 2.3.

1 Like

This topic was automatically closed 7 days after the last reply. New replies are no longer allowed.