Tensorflow error with BERT model selector

Partner Extensions
1

Hello,

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.

image
image1467×323 94.5 KB

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?

Regards,
Ben

2

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

1 Like
3

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