Differentiated Tensors not Used in Graph

[programehr]

Hi, I've used a modified version of your code. The modification is supposed to be just some refactors (assuming I've made no mistakes). I ran it on MNIST and a simple network, and it worked. But when I tried to use it on Cifar10 with Resnet18_comp I got the following error (line 61 of hypergrad.py):

One of the differentiated Tensors appears to not have been used in the graph. Set allow_unused=True if this is the desired behavior.

Do you know what the reason might be?