haha

2024-09-04 14:52:26 -05:00 · 2024-09-04 14:52:26 -05:00 · d319d33368
commit d319d33368
parent 619369236b
1 changed files with 1 additions and 1 deletions
--- 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,