From b4c392ed2f30cfed2224d431cf4c69e4f0f04c93 Mon Sep 17 00:00:00 2001
From: asistradition <asistradition@users.noreply.github.com>
Date: Thu, 11 Jul 2024 15:07:47 -0400
Subject: [PATCH] Cleanup training logic a bit

---
 supirfactor_dynamical/models/biophysical_model.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/supirfactor_dynamical/models/biophysical_model.py b/supirfactor_dynamical/models/biophysical_model.py
index 88b473e..6df21b9 100644
--- 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