Commit 7f02a3cd authored by Igor Saprykin's avatar Igor Saprykin Committed by TensorFlower Gardener
Browse files

Re-order arguments on the replicated model_fn. 

This supports the use cases that call Estimator's model_fn via positional arguments.

The right order is defined by Estimator as follows: https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/estimator/estimator.py#L102.

PiperOrigin-RevId: 175624067
parent f9e3e8d8
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment