Describe the bug
I try to run tensorboardX/examples/demo_graph.py for jupyter notebook (launched by anaconda navigator) and I get the error seen at Additional context.

I just copy paste the code to notebook from Github.

Minimal runnable code to reproduce the behavior
class SimpleModel(nn.Module):
def init(self):
super(SimpleModel, self).init()

Hi I would like to propose a better implementation for 'test_indices':

We can remove the unneeded np.array casting:

Cleaner/New:
test_indices = list(set(range(len(texts))) - set(train_indices))

Old:
test_indices = np.array(list(set(range(len(texts))) - set(train_indices)))

When I delete the lib.utils. ,the test.py is right but the demo.py is wrong , however,when I use the lib.utils. , the demo.py is right but the test.py is wrong !!

why? why

Thanks to @wilcoschoneveld (see #41) for this.

I explain a lot of this in the README, but it would be better as part of the visualization, and make it easier to count and trace stuff back to your code (e.g. Huh, that filter looks interesting, but there are 256 of them, and I'm not going to count all of them to figure out which one it actually is).

I'm not sure how all this would work, give

When a TF serving model returns a failure instead of a prediction, pass the failure string to the front-end for display instead of the generic http error we see now.