diff options
Diffstat (limited to 'tensorflow/contrib/factorization/python/ops/factorization_ops.py')
-rw-r--r-- | tensorflow/contrib/factorization/python/ops/factorization_ops.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/factorization/python/ops/factorization_ops.py b/tensorflow/contrib/factorization/python/ops/factorization_ops.py index 167b442dbc..b853652629 100644 --- a/tensorflow/contrib/factorization/python/ops/factorization_ops.py +++ b/tensorflow/contrib/factorization/python/ops/factorization_ops.py @@ -128,7 +128,7 @@ class WALSModel(object): # model_init_op is passed to Supervisor. Chief trainer runs it. Other # trainers wait. - sv = tf.Supervisor(is_chief=is_chief, + sv = tf.train.Supervisor(is_chief=is_chief, ..., init_op=tf.group(..., model_init_op, ...), ...) ... |