From 6cc00ada595cb5bcd04c8b33f99d3c9cf578e52e Mon Sep 17 00:00:00 2001
From: Polina Turishcheva <49938348+pollytur@users.noreply.github.com>
Date: Fri, 8 Mar 2024 14:44:40 +0100
Subject: [PATCH] Update neuralpredictors/layers/encoders/encoder3d.py

Co-authored-by: Max Burg <27483850+MaxFBurg@users.noreply.github.com>
---
 neuralpredictors/layers/encoders/encoder3d.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/neuralpredictors/layers/encoders/encoder3d.py b/neuralpredictors/layers/encoders/encoder3d.py
index 84bca35d..8042575f 100644
--- a/neuralpredictors/layers/encoders/encoder3d.py
+++ b/neuralpredictors/layers/encoders/encoder3d.py
@@ -15,7 +15,7 @@ def __init__(self, core, readout, readout_nonlinearity, elu_xshift, elu_yshift):
     def forward(self, x, data_key=None, pupil_center=None, trial_idx=None, shift=None, detach_core=False, **kwargs):
         out_core = self.core(x)
         if detach_core:
-            x = x.detach()
+            out_core = out_core.detach()
 
         if self.shifter:
             if pupil_center is None: