Issue with gradient computation

[anirbanl]

On running the following line of code in main.py, i am getting RuntimeError:
gradients, label_index = calculate_outputs_and_gradients([img], model, None, args.cuda)

Error trace:

RuntimeError Traceback (most recent call last)
in ()
5 img = img[:, :, (2, 1, 0)]
6 # calculate the gradient and the label index
----> 7 gradients, label_index = calculate_outputs_and_gradients([img], model, None, args.cuda)

/dccstor/cssblr/anirban/integrated-gradient-pytorch/utils.pyc in calculate_outputs_and_gradients(inputs, model, target_label_idx, cuda)
21 # clear grad
22 model.zero_grad()
---> 23 output.backward()
24 gradient = input.grad.detach().cpu().numpy()[0]
25 gradients.append(gradient)

/dccstor/anirlaha1/deep/apr2018/lib/python2.7/site-packages/torch/tensor.pyc in backward(self, gradient, retain_graph, create_graph)
100 products. Defaults to False.
101 """
--> 102 torch.autograd.backward(self, gradient, retain_graph, create_graph)
103
104 def register_hook(self, hook):

/dccstor/anirlaha1/deep/apr2018/lib/python2.7/site-packages/torch/autograd/init.pyc in backward(tensors, grad_tensors, retain_graph, create_graph, grad_variables)
88 Variable._execution_engine.run_backward(
89 tensors, grad_tensors, retain_graph, create_graph,
---> 90 allow_unreachable=True) # allow_unreachable flag
91
92

RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation

Please help. Thanks.

[tumble-weed]

faced same problem, changing the architecture to resnet152 worked:
python main.py --cuda --img='01.jpg' --model-type resnet152

so might have something to do with the inception architecture.