attention maps

[boop-yyt]

Hi, I want to generate segmentation maps from my own data , but after download the checkpoint and config file ,
I modified the following code line 413, in vision_transformer.py:

# before
    w = np.load(checkpoint_path)
# after
    w = np.load(checkpoint_path, allow_pickle=True)

I got this error:

OSError: Failed to interpret file '/xxxx/.cache/torch/hub/checkpoints/Ti_16-i21k-300ep-lr_0.001-aug_none-wd_0.03-do_0.0-sd_0.0--imagenet2012-steps_20k-lr_0.03-res_384.npz' as a pickle

when I check this .npz, most of words was garbled.