iurada / px-ntk-pruning

Official repository of our work "Finding Lottery Tickets in Vision Models via Data-driven Spectral Foresight Pruning" accepted at CVPR 2024
https://iurada.github.io/PX
21 stars 4 forks source link

KeyError:'state_dict' #3

Closed vankari closed 4 months ago

vankari commented 4 months ago

i downloaded the .pth from the url provided in "model/imagenet_resnet.py" then run the "cifar100_pretrain.sh",and encountered key error. image

iurada commented 4 months ago

Hi!

I guess you're trying to run the experiments on CIFAR-100 starting from the MoCoV2 pretrained checkpoint. In this case you should download it from eg. https://github.com/facebookresearch/moco.

Then, place the checkpoint in a folder and set the path in the --experiment_args option within the launch script:

Schermata 2024-07-23 alle 10 37 35
vankari commented 4 months ago

thank you!