From fec014e92d83106b3d8070d6df51fa3d6838d570 Mon Sep 17 00:00:00 2001
From: Lin Manhui <mhlin425@whu.edu.cn>
Date: Mon, 31 Jul 2023 15:13:11 +0800
Subject: [PATCH] Suppress scalar warnings (#10509)

---
 tools/program.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/tools/program.py b/tools/program.py
index a1a78090..511ee9dd 100755
--- a/tools/program.py
+++ b/tools/program.py
@@ -346,7 +346,10 @@ def train(config,
                 lr_scheduler.step()
 
             # logger and visualdl
-            stats = {k: v.numpy().mean() for k, v in loss.items()}
+            stats = {
+                k: float(v) if v.shape == [] else v.numpy().mean()
+                for k, v in loss.items()
+            }
             stats['lr'] = lr
             train_stats.update(stats)
 
-- 
GitLab