From 9dcc6bdb9750e8d4fcda13d7a32fd6a66e36c03c Mon Sep 17 00:00:00 2001 From: 51616 Date: Wed, 15 Jan 2025 18:20:09 +0000 Subject: [PATCH] add back gen metrics --- hyperlora/watcher.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/hyperlora/watcher.py b/hyperlora/watcher.py index a78244f..a0c8a2b 100644 --- a/hyperlora/watcher.py +++ b/hyperlora/watcher.py @@ -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}")