Skip to content

GitLab

Commit 8f6e2547 authored by guptapriya's avatar guptapriya Committed by guptapriya
Browse files

use tf.keras.losses instead of tf.losses

parent 6da769b1
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/recommendation/ncf_keras_main.py
View file @ 8f6e2547
...@@ -309,7 +309,7 @@ def run_ncf(_): ...@@ -309,7 +309,7 @@ def run_ncf(_):
epsilon=params["epsilon"]) epsilon=params["epsilon"])
if params["keras_use_ctl"]: if params["keras_use_ctl"]:
loss_object = tf.losses.SparseCategoricalCrossentropy( loss_object = tf.keras.losses.SparseCategoricalCrossentropy(
reduction=tf.keras.losses.Reduction.SUM, reduction=tf.keras.losses.Reduction.SUM,
from_logits=True) from_logits=True)
train_input_iterator = strategy.make_dataset_iterator(train_input_dataset) train_input_iterator = strategy.make_dataset_iterator(train_input_dataset)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment