Tensorboard Metrics (#22492)

nataliakliushkina · facebook-github-bot · commit 25b69997c309 · 2019-07-16T16:18:00.000-07:00
Summary: Pull Request resolved: #22492 Collect metrics about Tensorboard usage [internal] fbcode/pytorch/tensorboardX/tensorboardX/writer.py [OSS] fbcode/caffe2/torch/utils/tensorboard/writer.py Tensorboard Ondemand https://fb.quip.com/JRvqAKtzgy6z Reviewed By: dzhulgakov Differential Revision: D16105544 fbshipit-source-id: de14e6ec781889e367a6eba39fc777f707628263
diff --git a/torch/utils/tensorboard/writer.py b/torch/utils/tensorboard/writer.py
@@ -8,6 +8,7 @@
 import os
 import six
 import time
+import torch
 
 from tensorboard.compat.proto.event_pb2 import SessionLog
 from tensorboard.compat.proto.event_pb2 import Event
@@ -201,6 +202,7 @@ def __init__(self, log_dir=None, comment='', purge_step=None, max_queue=10,
             # folder location: runs/May04_22-14-54_s-MacBook-Pro.localLR_0.1_BATCH_16/
 
         """
+        torch._C._log_api_usage_once("tensorboard.create.summarywriter")
         if not log_dir:
             import socket
             from datetime import datetime
@@ -643,7 +645,6 @@ def add_graph(self, model, input_to_model=None, verbose=False, **kwargs):
         """
         if hasattr(model, 'forward'):
             # A valid PyTorch model should have a 'forward' method
-            import torch
             from distutils.version import LooseVersion
             if LooseVersion(torch.__version__) >= LooseVersion("0.3.1"):
                 pass
