Estimator for DeepRec is based on TensorFlow Estimator, which is high-level TensorFlow API that greatly simplifies machine learning programming. Estimators encapsulate training, evaluation, prediction, and exporting for your model.

New features in DeepRec Estimator:

• Support GRPC++ for large scale distributed training in parameter server mode.
• Support StarServer for large scale distributed training in parameter server mode.

CPU Dev Docker

GPU(cuda11.6) Dev Docker

Develop Branch：master, Latest Release Branch: deeprec2302

Build Package Builder

bazel build //tensorflow_estimator/tools/pip_package:build_pip_package

Build Package

bazel-bin/tensorflow_estimator/tools/pip_package/build_pip_package /tmp/estimator_whl

Installing DeepRec will install the native tensorflow-estimator by default, please reinstall the compiled Estimator.

• GRPC++
• StarServer

See TensorFlow Estimator getting started guide for an introduction to the Estimator APIs.