From f1e022a08c9aefcdd671a26eb9539defb28b5161 Mon Sep 17 00:00:00 2001 From: litagin02 Date: Sun, 26 May 2024 08:27:39 +0900 Subject: [PATCH] Fix tb logging --- train_ms.py | 2 +- train_ms_jp_extra.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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, )