请问backbones中rest可视化操作

[Idea89560041]

目前demo中有resnet50和vgg19两个可视化操作，请问rest的操作是怎样的，谢谢

[wofmanaf]

将此行groupcam.py#L34替换为

b, n, k = activations.size()
u, v = int(n**0.5), int(n**0.5)
gradients = gradients.permute(0, 2, 1).reshape(b, k, u, v)
activations = activations.permute(0, 2, 1).reshape(b, k, u, v)

Target layer for ResT is 'norm'