diff --git a/tests/data/models/min_tracks_2node.UNet.bottomup_multiclass/initial_config.json b/tests/data/models/min_tracks_2node.UNet.bottomup_multiclass/initial_config.json
index 96da29b77..2ae0e925c 100644
--- a/tests/data/models/min_tracks_2node.UNet.bottomup_multiclass/initial_config.json
+++ b/tests/data/models/min_tracks_2node.UNet.bottomup_multiclass/initial_config.json
@@ -128,7 +128,7 @@
         "runs_folder": "models",
         "tags": [],
         "save_visualizations": false,
-        "keep_viz_images": true,
+        "keep_viz_images": false,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,
diff --git a/tests/data/models/min_tracks_2node.UNet.bottomup_multiclass/training_config.json b/tests/data/models/min_tracks_2node.UNet.bottomup_multiclass/training_config.json
index ba462849a..7b6f817aa 100644
--- a/tests/data/models/min_tracks_2node.UNet.bottomup_multiclass/training_config.json
+++ b/tests/data/models/min_tracks_2node.UNet.bottomup_multiclass/training_config.json
@@ -191,7 +191,7 @@
         "runs_folder": "models",
         "tags": [],
         "save_visualizations": false,
-        "keep_viz_images": true,
+        "keep_viz_images": false,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,
diff --git a/tests/data/models/minimal_instance.UNet.bottomup/initial_config.json b/tests/data/models/minimal_instance.UNet.bottomup/initial_config.json
index 9b3e48b5e..68e4f894e 100644
--- a/tests/data/models/minimal_instance.UNet.bottomup/initial_config.json
+++ b/tests/data/models/minimal_instance.UNet.bottomup/initial_config.json
@@ -127,7 +127,7 @@
         "runs_folder": "models",
         "tags": [],
         "save_visualizations": false,
-        "keep_viz_images": true,
+        "keep_viz_images": false,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,
diff --git a/tests/data/models/minimal_instance.UNet.bottomup/training_config.json b/tests/data/models/minimal_instance.UNet.bottomup/training_config.json
index 488f90496..e3bfbc5f8 100644
--- a/tests/data/models/minimal_instance.UNet.bottomup/training_config.json
+++ b/tests/data/models/minimal_instance.UNet.bottomup/training_config.json
@@ -192,7 +192,7 @@
         "runs_folder": "models",
         "tags": [],
         "save_visualizations": false,
-        "keep_viz_images": true,
+        "keep_viz_images": false,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,
diff --git a/tests/data/models/minimal_instance.UNet.centered_instance/initial_config.json b/tests/data/models/minimal_instance.UNet.centered_instance/initial_config.json
index 23c5af795..f4914aae4 100644
--- a/tests/data/models/minimal_instance.UNet.centered_instance/initial_config.json
+++ b/tests/data/models/minimal_instance.UNet.centered_instance/initial_config.json
@@ -119,7 +119,7 @@
         "runs_folder": "models",
         "tags": [],
         "save_visualizations": false,
-        "keep_viz_images": true,
+        "keep_viz_images": false,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,
diff --git a/tests/data/models/minimal_instance.UNet.centered_instance/training_config.json b/tests/data/models/minimal_instance.UNet.centered_instance/training_config.json
index c7e52a78e..e747f6862 100644
--- a/tests/data/models/minimal_instance.UNet.centered_instance/training_config.json
+++ b/tests/data/models/minimal_instance.UNet.centered_instance/training_config.json
@@ -179,7 +179,7 @@
         "runs_folder": "models",
         "tags": [],
         "save_visualizations": false,
-        "keep_viz_images": true,
+        "keep_viz_images": false,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,
diff --git a/tests/data/models/minimal_instance.UNet.centroid/initial_config.json b/tests/data/models/minimal_instance.UNet.centroid/initial_config.json
index c1d1d2fc5..977654b2e 100644
--- a/tests/data/models/minimal_instance.UNet.centroid/initial_config.json
+++ b/tests/data/models/minimal_instance.UNet.centroid/initial_config.json
@@ -118,7 +118,7 @@
         "runs_folder": "models",
         "tags": [],
         "save_visualizations": false,
-        "keep_viz_images": true,
+        "keep_viz_images": false,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,
diff --git a/tests/data/models/minimal_instance.UNet.centroid/training_config.json b/tests/data/models/minimal_instance.UNet.centroid/training_config.json
index 9fa55ba0c..02e9683e1 100644
--- a/tests/data/models/minimal_instance.UNet.centroid/training_config.json
+++ b/tests/data/models/minimal_instance.UNet.centroid/training_config.json
@@ -175,7 +175,7 @@
         "runs_folder": "models",
         "tags": [],
         "save_visualizations": false,
-        "keep_viz_images": true,
+        "keep_viz_images": false,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,