From 65482b55a695ef317a298197e7f78eb977fa38e1 Mon Sep 17 00:00:00 2001 From: Carl Philipp Klemm Date: Tue, 7 May 2024 15:27:59 +0200 Subject: [PATCH] actually print eval response --- train_dynamic.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train_dynamic.py b/train_dynamic.py index 818d730..6425de9 100644 --- a/train_dynamic.py +++ b/train_dynamic.py @@ -78,7 +78,7 @@ def evaluate(model: DyntrainModel, tokenizer, attention_mask = torch.ones(input_ids.shape, device=model.devices[0], requires_grad=False) outputs = model.generate(input_ids, attention_mask=attention_mask, do_sample=True, temperature=1, max_new_tokens=100) response_decoded = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0] - print(f"Eval generation: response_decoded") + print(f"Eval generation: {response_decoded}") log_writer.add_text("Text/Eval", response_decoded, globalstep)