fix #86

lanpa · Feb 16, 2018 · 89322aa · 89322aa
1 parent 49ce85f
commit 89322aa
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/docs/conf.py b/docs/conf.py
@@ -99,7 +99,7 @@
 # Add any paths that contain custom static files (such as style sheets) here,
 # relative to this directory. They are copied after the builtin static files,
 # so a file named "default.css" will overwrite the builtin "default.css".
-html_static_path = ['_static']
+# html_static_path = ['_static']
 
 
 # -- Options for HTMLHelp output ------------------------------------------

diff --git a/tensorboardX/graph.py b/tensorboardX/graph.py
@@ -4,7 +4,6 @@
 from .src.attr_value_pb2 import AttrValue
 from .src.tensor_shape_pb2 import TensorShapeProto
 
-import torch
 from distutils.version import LooseVersion
 
 
@@ -13,6 +12,7 @@ def replace(name, scope):
 
 
 def parse(graph):
+    import torch
     scope = {}
     for n in graph.nodes():
         inputs = [i.uniqueName() for i in n.inputs()]
@@ -47,6 +47,7 @@ def parse(graph):
 
 
 def graph(model, args, verbose=False):
+    import torch
     with torch.onnx.set_training(model, False):
         trace, _ = torch.jit.trace(model, args)
     if LooseVersion(torch.__version__) >= LooseVersion("0.4"):