From 06edb9e2121be1288980c93434adeccbe039e4cd Mon Sep 17 00:00:00 2001
From: Alan <41682961+alan-cooney@users.noreply.github.com>
Date: Tue, 30 Jan 2024 09:55:58 -0300
Subject: [PATCH] Fix loss cloning

---
 sparse_autoencoder/train/pipeline.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/sparse_autoencoder/train/pipeline.py b/sparse_autoencoder/train/pipeline.py
index 99754672..53e3e3c2 100644
--- a/sparse_autoencoder/train/pipeline.py
+++ b/sparse_autoencoder/train/pipeline.py
@@ -498,7 +498,7 @@ def run_pipeline(
         )
 
         # Get the loss fn
-        loss_fn = self.autoencoder.loss_metric.clone()
+        loss_fn = self.loss.clone()
         loss_fn.keep_batch_dim = True
 
         with tqdm(