Do you support TF Estimator

Do you support TF Estimator? tf.estimator.Estimator  |  TensorFlow Core v2.6.1

1 Like

Hi Jonathan,

Yes we do support tf.estimator

Please take a look at some examples in our Model References

(e.g. Model-References/TensorFlow/computer_vision/efficientdet/horovod_estimator/estimator.py).

Please also notice the comment specifying the required version in section 3.4.2.1 of the install guide in case you are not going to use HabanaLabs TF docker and want to install TF yourself.

https://docs.habana.ai/en/v1.1.0/Installation_Guide/GAUDI_Installation_Guide.html#base-installation-single-node

Best regards,

-Leon

3 Likes