update docs

phuicy · Oct 9, 2018 · a2e6854 · a2e6854
1 parent 391a4ae
commit a2e6854
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -27,7 +27,7 @@ or build from source:
 
 ## Example
 
-* Run the demo script: `python demo.py`
+* Run the demo script: `python examples/demo.py`
 * Use TensorBoard with `tensorboard --logdir runs`  (needs to install TensorFlow)
 
 ```python
@@ -94,7 +94,8 @@ writer.close()
 
 ## Tweaks
 
-To add more ticks for the slider (show more image history), check https://github.com/lanpa/tensorboardX/issues/44
+To add more ticks for the slider (show more image history), check https://github.com/lanpa/tensorboardX/issues/44 or 
+https://github.com/tensorflow/tensorboard/pull/1138
 
 ## Reference
 

diff --git a/examples/demo_graph.py b/examples/demo_graph.py
@@ -59,7 +59,7 @@ def forward(self, x):
     w.add_graph(model, (dummy_input, ))
 
 
-dummy_input = Variable(torch.rand(1, 3, 224, 224))
+dummy_input = torch.Tensor(1, 3, 224, 224)
 
 with SummaryWriter(comment='alexnet') as w:
     model = torchvision.models.alexnet()