Update neuralpredictors/layers/encoders/encoder3d.py

Co-authored-by: Max Burg <[email protected]>
sinzlab · Mar 8, 2024 · 6cc00ad · 6cc00ad
1 parent 177cd97
commit 6cc00ad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: