search

LiQiufu / WaveCNet

the code for WaveCNet
123 stars 16 forks source link

About visual_results Issues #3

Open inkplatform opened 3 years ago

inkplatform commented 3 years ago

I am very sorry because I did not find any code for visual_results. If there is such code in the project, could you show it out for me?

LiQiufu commented 3 years ago

Sorry, I didn't upload the code for visualizing the feature maps. Your difficulty may be how to get the internal feature maps. Recently, I find a easy way to get them, using "register_forward_hook" https://pytorch.org/docs/stable/generated/torch.nn.Module.html?highlight=register_forward_hook#torch.nn.Module.register_forward_hook