diff --git a/references/detection/train_tensorflow.py b/references/detection/train_tensorflow.py
index bed759187..0b61c4c29 100644
--- a/references/detection/train_tensorflow.py
+++ b/references/detection/train_tensorflow.py
@@ -132,7 +132,7 @@ def fit_one_epoch(model, train_loader, batch_transforms, optimizer, amp=False):
         if current_progress - last_progress > interval_progress:
             send_on_slack(str(pbar))
             last_progress = int(current_progress)
-    send_on_slack(f"Final training loss: {train_loss.item():.6}")
+    send_on_slack(f"Final training loss: {train_loss.numpy():.6}")
 
 
 def evaluate(model, val_loader, batch_transforms, val_metric):