Skip to content

Commit e976927

Browse files
committed
Merge: [WideAndDeep/TF2] Fix DLLogger bug
2 parents e3b7e01 + 5668c36 commit e976927

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

  • TensorFlow2/Recommendation/WideAndDeep/trainer/utils

TensorFlow2/Recommendation/WideAndDeep/trainer/utils/evaluator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -265,7 +265,7 @@ def eval(self, step):
265265
eval_data["streaming_map_val"] = np.around(map_value, 4)
266266

267267
for metric_name, metric in zip(self.metric_names, hand_reduced_metrics):
268-
eval_data[metric_name] = np.around(metric.result().numpy(), 4)
268+
eval_data[metric_name] = np.around(metric.result().numpy().astype(np.float64), 4)
269269

270270
self.log(eval_data, current_step)
271271

0 commit comments

Comments
 (0)