pos_embed key missing from checkpoints
schmitzmatheus opened this issue · 1 comments
schmitzmatheus commented
On main.py (https://github.com/Sense-X/UniFormer/blob/main/image_classification/main.py#L276), we try to access key pos_embed on the loaded checkpoint, but for the checkpoints provided, the key is missing and so the code fails.
This happens when running run.sh (https://github.com/Sense-X/UniFormer/blob/main/image_classification/exp/uniformer_small/run.sh) with the provided checkpoints for fineturning (--finetune ).
schmitzmatheus commented
I just saw #88