mrq/vall-e
5
9
Fork 0
Code Issues 8 Pull Requests Packages Projects Releases Wiki Activity

haha

Browse Source
This commit is contained in:
mrq 2024-09-04 14:52:26 -05:00
parent 619369236b
commit d319d33368
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vall_e/engines/base.py
View File

@ -568,7 +568,7 @@ class Engines(dict[str, Engine]):
name.split("-")[0]: dict( name.split("-")[0]: dict(
**engine_stats, **engine_stats,
lr=engine.get_lr()[0], 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, loss_scale=loss_scale if loss_scale != 1 else None,
elapsed_time=elapsed_time, elapsed_time=elapsed_time,
engine_step=engine.global_step, engine_step=engine.global_step,