minor

justanhduc · Apr 27, 2022 · ee9be98 · ee9be98
1 parent 9d88105
commit ee9be98
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/neural_monitor/monitor.py b/neural_monitor/monitor.py
@@ -245,7 +245,9 @@ def standardize_image(img):
 
 
 def _convert_time_human_readable(t):
-    if t < 3600:
+    if t < 60:
+        time_unit = 's'
+    elif 60 <= t < 3600:
         time_unit = 'mins'
         t /= 60.
     elif 86400 > t >= 3600: