[Bug] vis_cam.py error

[a819411321]

Branch

main branch (mmpretrain version)

Describe the bug

Environment

use vis_cam.py to see featmap of swin-small_16xb64_in1k, but error. Error imshow "The input feature's length (192) minus num-extra-tokens (0) is 192, which is not a perfect square number. Please check if you used a wrong num-extra-tokens."

Other information

No response