About the visualization of the middle layer feature map

[a-die]

Hello, I now need to visualize the feature map of the middle layer to see if the network has learned the relevant information. I modified the network based on nnUNet, so I wonder if there is any way to visualize the feature map of the middle level based on nnUNet.