minor tblogger image logging bug fix

automl · Nov 22, 2023 · f6cecbf · f6cecbf
1 parent c9bc384
commit f6cecbf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/neps/plot/tensorboard_eval.py b/src/neps/plot/tensorboard_eval.py
@@ -306,7 +306,7 @@ def _write_image_config(
         if resize_images is None:
             resize_images = [32, 32]
 
-        if tblogger.current_epoch and tblogger.current_epoch % counter == 0:
+        if tblogger.current_epoch >= 0 and tblogger.current_epoch % counter == 0:
             # Log every multiple of "counter"
 
             if num_images > len(image):