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

ugh

Browse Source
This commit is contained in:
mrq 2025-03-06 17:19:27 -06:00
parent 2dd80a03ff
commit 91ede71cf0
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/train.py
View File

@ -251,7 +251,7 @@ def run_eval(engines, eval_name, dl, args=None):
} }
try: try:
for engine in engines: for name, engine in engines.items():
if engine.wandb is not None: if engine.wandb is not None:
engine.wandb.log({ engine.wandb.log({
f'{eval_name}.loss.mstft': stats['loss'], f'{eval_name}.loss.mstft': stats['loss'],