Make protocol used in estimator customizable.
Example code as follow:
config = tf.estimator.RunConfig(protocol='grpc+verbs')
nn = tf.estimator.Estimator(model_fn=model_fn,
model_dir=model_dir,
params=params,
config=config)
Loading
Please sign in to comment