How do I display computation graph in tensorboard from pytorch?
I logged my output images to tensorboard as follows:
import tensorflow as tf
self.tf = tf
self.writer = tf.summary.FileWriter('path/to/log/dir')
img_summaries = []
...
for label, image_numpy in visuals.items():
    try:
      s = StringIO()
   except:
      s = BytesIO()
   scipy.misc.toimage(image_numpy).save(s, format="jpeg")
   img_sum = self.tf.Summary.Image(encoded_image_string=s.getvalue(), height=image_numpy.shape[0], width=image_numpy.shape[1])
   img_summaries.append(self.tf.Summary.Value(tag=label, image=img_sum))
   summary = self.tf.Summary(value=img_summaries)
   self.writer.add_summary(summary, step)

But when I try to add graph to tensorboard:
writer.add_graph(model, loss_1)
I get this error:
raise TypeError("The passed graph must be an instance of `Graph` "
TypeError: The passed graph must be an instance of `Graph` or the deprecated `GraphDef`

>>> print(torch.__version__)
0.3.0.post4
>>> print(tensorflow.__version__)
1.4.1

What am I doing wrong here? Any help is appreciated. Thank you.