From 49c8e8e7b4b724ff9bfb77bc12161eebdcc44ca0 Mon Sep 17 00:00:00 2001 From: 51616 Date: Tue, 22 Apr 2025 10:48:23 +0000 Subject: [PATCH] rename l1 norm log name --- src/ctx_to_lora/training_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/ctx_to_lora/training_utils.py b/src/ctx_to_lora/training_utils.py index 6b9c0a1..23016bb 100644 --- a/src/ctx_to_lora/training_utils.py +++ b/src/ctx_to_lora/training_utils.py @@ -76,7 +76,7 @@ class ModulatedModelTrainer(Trainer): self.args.logging_strategy == IntervalStrategy.STEPS and self.state.global_step % self.state.logging_steps == 0 ): - self.log({"ce_loss": loss.item(), "gen_lora_reg_loss": l1_norm.item()}) + self.log({"ce_loss": loss.item(), "gen_lora_l1_norm": l1_norm.item()}) loss += self.gen_lora_l1_reg_coef * l1_norm #####