Move logit_fn from tf.contrib.learn to tf.contrib.estimator. Also, logit_fn's...
Move logit_fn from tf.contrib.learn to tf.contrib.estimator. Also, logit_fn's can now return a list of Tensors (for compatibility with MultiHead) -- canned dnn logit_fn adjusted accordingly. PiperOrigin-RevId: 169977733
Loading
Please sign in to comment