From d319d3336823cf1808358b143fb51e68932dd996 Mon Sep 17 00:00:00 2001
From: mrq <mrq@ecker.tech>
Date: Wed, 4 Sep 2024 14:52:26 -0500
Subject: [PATCH] haha

---
 vall_e/engines/base.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vall_e/engines/base.py b/vall_e/engines/base.py
index a077eaf..7fb04fd 100755
--- a/vall_e/engines/base.py
+++ b/vall_e/engines/base.py
@@ -568,7 +568,7 @@ class Engines(dict[str, Engine]):
 						name.split("-")[0]: dict(
 							**engine_stats,
 							lr=engine.get_lr()[0],
-							grad_norm=grad_norm,
+							grad_norm=grad_norm.item() if isinstance( grad_norm, torch.Tensor ) else grad_norm,
 							loss_scale=loss_scale if loss_scale != 1 else None,
 							elapsed_time=elapsed_time,
 							engine_step=engine.global_step,