adding new metric loss_batch from merlin models (#727)

NVIDIA-Merlin · Nov 21, 2022 · e334d75 · e334d75
1 parent 04e4bcb
commit e334d75
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/unit/examples/test_scaling_criteo_merlin_models.py b/tests/unit/examples/test_scaling_criteo_merlin_models.py
@@ -59,6 +59,7 @@ def test_func():
                 "precision",
                 "recall",
                 "regularization_loss",
+                "loss_batch",
             ]
         )
         assert os.path.isfile("/tmp/output/criteo/dlrm/saved_model.pb")