bug fix and run desc in tensorboard

2026-09-09 11:16:00 +08:00 · 2020-03-10 22:38:51 +01:00
parent 3472a41255
commit 2a15e39166
3 changed files with 10 additions and 4 deletions
@@ -75,3 +75,6 @@ class Logger(object):

    def tb_test_figures(self, step, figures):
        self.dict_to_tb_figure("TestFigures", figures, step)
+
+    def tb_add_text(self, title, text, step):
+        self.writer.add_text(title, text, step)