update model

cf67f606 · 赵威 · baa58b4f · cf67f606
Commit cf67f606 authored Jul 22, 2020 by 赵威
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

model.py src/models/esmm/model.py +1 -1

No files found.
--- a/src/models/esmm/model.py
+++ b/src/models/esmm/model.py
@@ -72,7 +72,7 @@ def esmm_model_fn(features, labels, mode, params):
            tf.compat.v1.summary.scalar("ctcvr_auc", ctcvr_auc[1])
            return tf.estimator.EstimatorSpec(mode, loss=loss, eval_metric_ops=metrics)
        train_op = optimizer.minimize(loss, global_step=tf.compat.v1.train.get_global_step())
-        res = tf.estimator.EstimatorSpec(mode, loss=loss, train_op=train_op, eval_metric_ops=metrics)
+        res = tf.estimator.EstimatorSpec(mode, loss=loss, train_op=train_op)
        return res