refactor

2020-12-01 14:06:29 +08:00
parent 5118d07b0f
commit 0c1389853a
246 changed files with 186 additions and 3834 deletions
@@ -1,14 +0,0 @@
-from utils.util import to_np
-
-
-def histogram_logger(tb_writer, iter_index, net=None):
-    if net is not None:
-        for tag, value in net.named_parameters():
-            tag = tag.replace(".", "/")
-            tb_writer.add_histogram(tag, to_np(value), iter_index)
-            tb_writer.add_histogram(tag + "/grad", to_np(value.grad), iter_index)
-
-
-def value_logger(tb_writer, iter_index, loss_info):
-    for tag, value in loss_info.items():
-        tb_writer.add_scalar(tag, value, iter_index)