Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

Variable slice/index assignment graph breaking

See original GitHub issue

I have been facing an issue when trying to create a graph of a Module in which some Variables have slice assignment operations in them. I have reduced the problem to the following example, ignore the commented out V = x for now.

import torch
from torch.autograd import Variable
from tensorboardX import SummaryWriter

class DummyModule(torch.nn.Module):
    def forward(self, x):
        V = Variable(torch.Tensor(2, 2))
        V[0, 0] = x
        # V = x
        return torch.sum(V * 3)

x = Variable(torch.Tensor([1]), requires_grad=True)
r = DummyModule()(x)

w = SummaryWriter()
x = Variable(torch.Tensor([1]), requires_grad=True)
w.add_graph(DummyModule(), x, verbose=True)

The output from this is below, showing that the gradients are flowing all right, but the graph is not being connected. If I insert another input Variable and other operations in the Module, add_graph() works fine without throwing an error, but the graph show a disconnected input for x, so I suppose the nature of this error is that the only input Variable available is being interpreted as disconnected.

Variable containing:
[torch.FloatTensor of size (1,)]

Traceback (most recent call last):
  File "", line 21, in <module>
    w.add_graph(DummyModule(), x, verbose=True)
  File "/Users/filiped/anaconda/envs/pytorch0.4/lib/python3.6/site-packages/tensorboardX/", line 400, in add_graph
    self.file_writer.add_graph(graph(model, input_to_model, verbose))
  File "/Users/filiped/anaconda/envs/pytorch0.4/lib/python3.6/site-packages/tensorboardX/", line 44, in graph
    trace, _ = torch.jit.trace(model, args)
  File "/Users/filiped/anaconda/envs/pytorch0.4/lib/python3.6/site-packages/torch/jit/", line 251, in trace
    return TracedModule(f, nderivs=nderivs)(*args, **kwargs)
  File "/Users/filiped/anaconda/envs/pytorch0.4/lib/python3.6/site-packages/torch/nn/modules/", line 357, in __call__
    result = self.forward(*input, **kwargs)
  File "/Users/filiped/anaconda/envs/pytorch0.4/lib/python3.6/site-packages/torch/jit/", line 287, in forward
RuntimeError: /Users/filiped/pytorch/torch/csrc/jit/tracer.h:117: getTracingState: Assertion `state` failed.

Moreover, if you uncomment the line V = x and comment the line above it, so that no slice/index assign operation is performed, you get, as expected:

Variable containing:
[torch.FloatTensor of size (1,)]

graph(%0 : Float(1)) {
  %1 : UNKNOWN_TYPE = Constant[value={3}](), scope: DummyModule
  %2 : Float(1) = Mul[broadcast=1](%0, %1), scope: DummyModule
  %3 : Float() = Sum(%2), scope: DummyModule
  return (%3);

This was all executed in Pytorch 0.4

(Edits: Did a couple rounds of re-simplifying the example.)

Issue Analytics

  • State:open
  • Created 6 years ago
  • Comments:18 (10 by maintainers)

github_iconTop GitHub Comments

lanpacommented, May 8, 2018

update: still not working in pytorch 0.4 release + tensorboardX master. output of tensorboardX:

Error occurs, No graph saved
Checking if it's onnx problem...
Your model fails onnx too, please report to onnx team
lanpacommented, Mar 3, 2019

Interesting, I closed this because the onnx error is disappeared. Reopen it.

Read more comments on GitHub >

github_iconTop Results From Across the Web

ValueError: slice index 1 of dimension 0 out of ... - GitHub
I am trying to optimise my Mask-RCNN model trained on coco weights and uses ResNet50 as backbone. I had the h5 weights, along...
Read more >
Sliced Variables - MATLAB & Simulink - MathWorks
A sliced variable is one whose value can be broken up into segments, or slices, which are then operated on separately by different...
Read more >
Debugging in TensorFlow - Towards Data Science
In machine learning, the task of debugging is complicated by the stochasticity ... that causes your training session to abruptly break down.
Read more >
Pylint features - Pylint 2.16.0-dev documentation
Else clause on loop without a break statement, remove the else and de-indent all the code inside it Loops should only have an...
Read more >
Slicing a list using a variable, in Python - Stack Overflow
With the assignments below you are still using the same type of slicing operations you show, but now with variables for the values....
Read more >

github_iconTop Related Medium Post

No results found

github_iconTop Related StackOverflow Question

No results found

github_iconTroubleshoot Live Code

Lightrun enables developers to add logs, metrics and snapshots to live code - no restarts or redeploys required.
Start Free

github_iconTop Related Reddit Thread

No results found

github_iconTop Related Hackernoon Post

No results found

github_iconTop Related Tweet

No results found

github_iconTop Related Post

No results found

github_iconTop Related Hashnode Post

No results found