From 1eab63ae31c0871115e39a508630bcd1fdcde436 Mon Sep 17 00:00:00 2001
From: Benjamin-Walker <ben.walker@ladeck.com>
Date: Thu, 28 Aug 2025 22:37:36 +0100
Subject: [PATCH] Updated normalisation

---
 train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/train.py b/train.py
index a119461..3561e50 100644
--- a/train.py
+++ b/train.py
@@ -42,7 +42,7 @@ def vfA_l2(block):
                 tensors.append(t)
     else:
         tensors = [lcde.vf_A_diag, lcde.vf_A_dense.weight]
-    return torch.sqrt(sum((t**2).sum() for t in tensors))
+    return sum((t**2).sum() ** 0.5 for t in tensors)
 
 
 def train_model(
@@ -239,7 +239,7 @@ def train_model(
 
                 out_filename = f"results_{task}_{model_name}_{time_str}.json"
 
-                out_path = os.path.join("result", out_filename)
+                out_path = os.path.join("results", out_filename)
 
                 # Gather all relevant info to save:
                 results_dict = {