Fix tb logging

2024-05-26 08:27:39 +09:00
parent 5fa6210176
commit f1e022a08c
2 changed files with 2 additions and 2 deletions
--- 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,
                )
--- 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,
                )