fixed #30

lanpa · Oct 12, 2017 · 83b90b7 · 83b90b7
1 parent 78e6912
commit 83b90b7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/demo.py b/demo.py
@@ -30,7 +30,7 @@
             x[i] = np.cos(freqs[n_iter//10]*np.pi*float(i)/float(sample_rate)) # sound amplitude should in [-1, 1]
         writer.add_audio('myAudio', x, n_iter)
         writer.add_text('Text', 'text logged at step:'+str(n_iter), n_iter)
-        writer.add_text('another Text', 'another text logged at step:'+str(n_iter), n_iter)
+        writer.add_text('markdown Text', '''a|b\n-|-\nc|d''', n_iter)
         for name, param in resnet18.named_parameters():
             writer.add_histogram(name, param, n_iter)
 

diff --git a/tensorboardX/summary.py b/tensorboardX/summary.py
@@ -206,6 +206,6 @@ def text(tag, text):
   import json
   PluginData = [SummaryMetadata.PluginData(plugin_name='text')]
   smd = SummaryMetadata(plugin_data=PluginData)
-  tensor = TensorProto(dtype='DT_STRING', string_val=[text.encode(encoding='utf_8')])
+  tensor = TensorProto(dtype='DT_STRING', string_val=[text.encode(encoding='utf_8')], tensor_shape=TensorShapeProto(dim=[TensorShapeProto.Dim(size=1)]))
   return Summary(value=[Summary.Value(node_name=tag, metadata=smd, tensor=tensor)])