Change tensorboardX to pytorch.utils.tensorboard (openvinotoolkit#332)

* Change tensorboardX to tensorboard * Add tensorboard version
vshampor · Dec 8, 2020 · 45dd0a9 · 45dd0a9
1 parent 1f6303e
commit 45dd0a9
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/examples/common/argparser.py b/examples/common/argparser.py
@@ -159,7 +159,7 @@ def get_common_argument_parser():
         "--log-dir",
         type=str,
         default='runs',
-        help="The directory where models and TensorboardX summaries"
+        help="The directory where models and Tensorboard summaries"
              " are saved. Default: runs")
 
     parser.add_argument('--save-freq', default=5, type=int,

diff --git a/examples/common/utils.py b/examples/common/utils.py
@@ -28,7 +28,7 @@
 import torch.utils.data as data
 
 from examples.common.sample_config import SampleConfig
-from tensorboardX import SummaryWriter
+from torch.utils.tensorboard import SummaryWriter
 from texttable import Texttable
 import mlflow
 

diff --git a/setup.py b/setup.py
@@ -50,7 +50,7 @@ def find_version(*file_paths):
                     "graphviz==0.15",
                     "jsonschema==3.2.0",
                     "pydot==1.4.1",
-                    "tensorboardX==2.1",
+                    "tensorboard==2.4.0",
                     "jstyleson==0.0.2",
                     "matplotlib==3.3.3",
                     "numpy==1.19",