make some update

chaobrain · Aug 26, 2024 · b2fd817 · b2fd817
1 parent 472f8c2
commit b2fd817
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/fitting_simple_dendrite_model.py b/examples/fitting_simple_dendrite_model.py
@@ -123,7 +123,8 @@ def simulate(currents, params):
   hh.init_state()
 
   times = np.arange(0, currents.shape[0]) * bst.environ.get_dt()
-  vs, spks = bst.transform.for_loop(hh.step_run, times, currents)
+  # vs, spks = bst.transform.for_loop(hh.step_run, times, currents)
+  vs, spks = bst.transform.checkpointed_for_loop(hh.step_run, times, currents)
   return vs, spks