'builtin_function_or_method' object has no attribute 'grad_fn'

[Meimeiainaonao]

Thanks for your sharing! But I meet this question about

` File "/home/meichen/PycharmProjects/CBAM /cbam+resnet/MODELS/model_resnet.py", line 282, in make_dot(output.mean, params=dict(net.named_parameters())) File "/home/meichen/anaconda3/envs/reidbaseline/lib/python3.6/site-packages/torchviz/dot.py", line 37, in make_dot output_nodes = (var.grad_fn,) if not isinstance(var, tuple) else tuple(v.grad_fn for v in var)

AttributeError: 'builtin_function_or_method' object has no attribute 'grad_fn' `

Pls help. Thank you very much.

[VaticanCameos99]

It has to be output.mean(). You missed the braces.