fix

odulcy-mindee · Feb 28, 2024 · 4353faa · 4353faa
1 parent cabbf08
commit 4353faa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):