closes #80

lanpa · Feb 21, 2018 · 45a0626 · 45a0626
1 parent 3da4e2f
commit 45a0626
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/HISTORY.rst b/HISTORY.rst
@@ -1,6 +1,11 @@
 History
 =======
+1.1 (2018-02-21)
+-----------------
+* Supports pytorch 0.3.1 (hacky)
+
 1.0 (2018-01-18)
+-----------------
 * Supports graph (the pretty one)
 
 0.9 (2017-11-11)

diff --git a/demo_embedding.py b/demo_embedding.py
@@ -24,7 +24,7 @@ def forward(self, i):
         i = F.max_pool2d(i, 2)
         i = i.view(len(i), -1)
         i = self.fc1(i)
-        i = F.log_softmax(i)
+        i = F.log_softmax(i, dim=1)
         return i
 
 #get some random data around value
@@ -80,4 +80,4 @@ def get_data(value, shape):
 
 # tensorboard --logdir runs
 # you should now see a dropdown list with all the timestep,
-# last timestep should have a visible separation between the two classes
+# last timestep should have a visible separation between the two classes
diff --git a/setup.py b/setup.py
@@ -19,7 +19,7 @@
 
 setup(
     name='tensorboardX',
-    version='1.0',
+    version='1.1',
     description='TensorBoardX lets you watch Tensors Flow without Tensorflow',
     long_description= history,
     author='Tzu-Wei Huang',
@@ -46,5 +46,5 @@
     tests_require=test_requirements
 )
 
-# python setup.py sdist bdist_wheel --universal upload
+# python setup.py sdist bdist_wheel --universal
 # twine upload dist/*