apunkt/doc-to-lora
1
0
Fork 0
mirror of https://github.com/SakanaAI/doc-to-lora.git synced 2026-07-23 17:01:04 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

add back gen metrics

Browse source
This commit is contained in:
51616 2025-01-15 18:20:09 +00:00
parent 9671fde108
commit 9dcc6bdb97
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
hyperlora/watcher.py
View file

@ -110,8 +110,8 @@ if __name__ == "__main__":
}
wandb.init(**wandb_kwargs)
metrics = evaluate(file, args, split="validation", generative=False)
# gen_metrics = evaluate(file, args, split="validation", generative=True)
# metrics.update(gen_metrics)
gen_metrics = evaluate(file, args, split="validation", generative=True)
metrics.update(gen_metrics)
wandb.log(metrics, step=curstep)
wandb.finish()
print(f"Logged metrics: {metrics}")