diff --git a/train_ms.py b/train_ms.py
index 31670ac..6f416e7 100644
--- a/train_ms.py
+++ b/train_ms.py
@@ -795,7 +795,7 @@ def train_and_evaluate(
                 utils.summarize(
                     writer=writer,
                     global_step=global_step,
-                    images=image_dict,
+                    # images=image_dict,
                     scalars=scalar_dict,
                 )
 
diff --git a/train_ms_jp_extra.py b/train_ms_jp_extra.py
index 36cbb6e..a7c2d89 100644
--- a/train_ms_jp_extra.py
+++ b/train_ms_jp_extra.py
@@ -943,7 +943,7 @@ def train_and_evaluate(
                 utils.summarize(
                     writer=writer,
                     global_step=global_step,
-                    images=image_dict,
+                    # images=image_dict,
                     scalars=scalar_dict,
                 )