Cleanup training logic a bit

GreshamLab · Jul 11, 2024 · b4c392e · b4c392e
1 parent c64ac42
commit b4c392e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/supirfactor_dynamical/models/biophysical_model.py b/supirfactor_dynamical/models/biophysical_model.py
@@ -441,8 +441,7 @@ def _training_step(
             train_x,
             optimizer[2] if self._decay_optimize(epoch_num) else
             optimizer[0],
-            loss_function,
-            input_x=self._slice_data_and_forward(train_x)
+            loss_function
         )
 
         return loss, decay_loss