From 784f54ebfb3b72f01f1cc35b8c5e8fe005f22c4c Mon Sep 17 00:00:00 2001 From: Janghaeng Lee Date: Tue, 25 Apr 2023 15:20:25 +0000 Subject: [PATCH] Temporary workaround: Use TensorboardX summary writer for torch trainer Using torch.utils.tensorboard causes malloc segfault with TF=1.12 --- src/utils/torch/trainer.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/utils/torch/trainer.py b/src/utils/torch/trainer.py index b19f4cdb..65e13aa3 100644 --- a/src/utils/torch/trainer.py +++ b/src/utils/torch/trainer.py @@ -40,10 +40,10 @@ def dummycontext(): import datetime -try: - from torch.utils.tensorboard import SummaryWriter -except: - from tensorboardX import SummaryWriter +#try: +# from torch.utils.tensorboard import SummaryWriter +#except: +from tensorboardX import SummaryWriter from src.config import ComputeMode, Precision, ConvMode, ModeKind, DataFormatKind