attention map
wushunshun opened this issue · comments
Hello, how to print the channel-wise and pixel-wise feature weights map of the Group Structure output? Can it be printed in training the dataset or after training the dataset? Thank you very much!
You need to use 'register_forward_hook' function to get middle result of the model.