Update helper_functions.py

mrdbourke · Nov 14, 2021 · a7c2bda · a7c2bda
1 parent 0c9c132
commit a7c2bda
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/helper_functions.py b/helper_functions.py
@@ -14,4 +14,4 @@ def print_train_time(start, end, device=None):
         print(f"\nTrain time on {device}: {total_time:.3f} seconds")
     else:
         print(f"\nTrain time: {total_time:.3f} seconds")
-    return total_time
+    return round(total_time, 3)