TypeError: hook() takes 2 positional arguments but 3 were given

Question

I'm new to pytorch and I'm trying to use hook() and register_forward_pre_hook in my project

What I've tried is

def get_features_hook(module,input):
    print(input)

handle_feat = alexnet.features[0].register_forward_pre_hook(get_features_hook)


a = alexnet(input_data)

And I got belows error at a = alexnet(input_data)

TypeError: get_features_hook() takes 2 positional arguments but 3 were given

I've lost few hours on this problem and I just can't able to figure it out.

Anyone likes to help me?

---

With Shai's help, I tried his codes, and I got this

Conv2d(3, 64, kernel_size=(11, 11), stride=(4, 4), padding=(2, 2))
get_features_hook called with 2 args:
    arg of type Conv2d
    arg of type tuple

File "<input>", line 2, in get_features_hook
NameError: name 'args' is not defined

Answer 1

如果get_features_hook是你的内部定义torch.nn.Module ，应标注为@staticmethod ，否则self隐式传递给它

Answer 2

I got the same error. I've tried re-running the notebook and that solved it