fix timing bug

simplymathematics · Nov 29, 2023 · cf64d90 · cf64d90
1 parent 3eb34d7
commit cf64d90
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/deckard/base/model/model.py b/deckard/base/model/model.py
@@ -131,8 +131,10 @@ def __call__(self, data: list, model: object, library=None):
             raise NotImplementedError(f"Training library {library} not implemented")
         try:
             start = process_time_ns()
+            start_timestamp = time()
             model.fit(data[0], data[2], **trainer)
             end = process_time_ns() - start
+            end_timestamp = time()
         except np.AxisError:  # pragma: no cover
             from art.utils import to_categorical